二重リストの書き換え

Question

a=[[1], [6, 17], [0]]
b=[[0 for y in x] for x in a ]
#b=[[0], [0, 0], [0]]

bを[[0], [1, 2], [3]]のように中身を0からの整数のリストにするにはどうしたらいいでしょう．

Answer 1

[0 for y in x]と書いてしまうと全ての要素が0になってしまいますので、0の代わりに順序数を0, 1, 2, ...のように順番に列挙してくれるイテレーターを用い、それに対して次々にnext(iterator)として値を取り出す方法が思いつきます。

あらかじめaの全ての要素の個数の程度がわかっていれば、

例1:

a = [[1], [6, 17], [0]]
iterator = range(100)
b = [[next(iterator) for _ in e] for e in a]

などと書けます。一応許容範囲内のコードとは思いますが、「100というなんだかよくわからない数(プログラミングではマジックナンバーなんて言い方をします)」がコードに出現している点がイマイチです。

もうすこしきちんと書くならaの要素数を数えるべきでしょう。

例2:

a = [[1], [6, 17], [0]]
n = sum(map(len, a))
iterator = range(n)
b = [[next(iterator) for _ in e] for e in a]

しかしiteratorが「無限に数列を列挙してくれる」ならaの要素数を気にする必要はなくなりますので、rangeの代わりに自前でジェネレーターを定義する方法もアリだと思います。

例3:

a = [[1], [6, 17], [0]]

def ordinal():
    i = 0
    while True:
        yield i
        i += 1

iterator = ordinal()
b = [[next(iterator) for _ in e] for e in a]

ただ短いとは言え関数定義を伴うので例2ぐらいがいいのでしょうか・・・個人的には例3が好みです。

---

追記: LouiS0616さんがコメントくださったように、例3のordinalの汎用版が標準ライブラリー(itertools.count)にありました！ということで、単に次のように書くのが自然だと思います。

import itertools

a = [[1], [6, 17], [0]]
iterator = itertools.count()
b = [[next(iterator) for _ in e] for e in a]

Comments

[LouiS0616]

> 短いとは言え関数定義を伴うので

itertools.countが使えそうです。

[KSwordOfHaste]

「等差数列＝＞numpyのlinspaceやarange」という発想でしかコードを書いたことがありませんでした。countは本サイトの過去の回答で何度か見たはずですけど・・・頭に残ってませんでした。countを使うのが自然ですね！
勉強になりました。コメントありがとうございます。