Diary over Finite Fields

515ひかるが書き溜めたメモとコラムと雑記

リストのネストをいい感じに書きたいときの話

Pythonの話である。ある規則にそって、ネストしたリストを作りたいときがよくある。いやないよ、って思う人もいるかもしれないが、僕は仕事でよくやる。

どういう感じかというと、だいたい次のような感じ

[0 1 2 3] [1 2 3 4] [2 3 4 5] [3 4 5 6]
[4 5 6 7] [5 6 7 8] [6 7 8 9] [7 8 9 10]
[8 9 10 11] [9 10 11 12] [10 11 12 13] [11 12 13 14]
...

つまり、先頭から4つ*1とり、先頭を1つずらしてそこからさらに4つとり、を繰り返す。

これをPythonのリスト内包表記を使って書くとこうなる(てきとうに0~99までで書いた):

m = 4
l = [[j for j in range(i, i+m)] for i in range(100 - m + 1)]
print(l)

# 実行結果
# [[0, 1, 2, 3], [1, 2, 3, 4], [2, 3, 4, 5], [3, 4, 5, 6], [4, 5, 6, 7], [5, 6, 7, 8], [6, 7, 8, 9], [7, 8, 9, 10], [8, 9, 10, 11], [9, 10, 11, 12], [10, 11, 12, 13], [11, 12, 13, 14], [12, 13, 14, 15], [13, 14, 15, 16], [14, 15, 16, 17], [15, 16, 17, 18], [16, 17, 18, 19], [17, 18, 19, 20], [18, 19, 20, 21], [19, 20, 21, 22], [20, 21, 22, 23], [21, 22, 23, 24], [22, 23, 24, 25], [23, 24, 25, 26], [24, 25, 26, 27], [25, 26, 27, 28], [26, 27, 28, 29], [27, 28, 29, 30], [28, 29, 30, 31], [29, 30, 31, 32], [30, 31, 32, 33], [31, 32, 33, 34], [32, 33, 34, 35], [33, 34, 35, 36], [34, 35, 36, 37], [35, 36, 37, 38], [36, 37, 38, 39], [37, 38, 39, 40], [38, 39, 40, 41], [39, 40, 41, 42], [40, 41, 42, 43], [41, 42, 43, 44], [42, 43, 44, 45], [43, 44, 45, 46], [44, 45, 46, 47], [45, 46, 47, 48], [46, 47, 48, 49], [47, 48, 49, 50], [48, 49, 50, 51], [49, 50, 51, 52], [50, 51, 52, 53], [51, 52, 53, 54], [52, 53, 54, 55], [53, 54, 55, 56], [54, 55, 56, 57], [55, 56, 57, 58], [56, 57, 58, 59], [57, 58, 59, 60], [58, 59, 60, 61], [59, 60, 61, 62], [60, 61, 62, 63], [61, 62, 63, 64], [62, 63, 64, 65], [63, 64, 65, 66], [64, 65, 66, 67], [65, 66, 67, 68], [66, 67, 68, 69], [67, 68, 69, 70], [68, 69, 70, 71], [69, 70, 71, 72], [70, 71, 72, 73], [71, 72, 73, 74], [72, 73, 74, 75], [73, 74, 75, 76], [74, 75, 76, 77], [75, 76, 77, 78], [76, 77, 78, 79], [77, 78, 79, 80], [78, 79, 80, 81], [79, 80, 81, 82], [80, 81, 82, 83], [81, 82, 83, 84], [82, 83, 84, 85], [83, 84, 85, 86], [84, 85, 86, 87], [85, 86, 87, 88], [86, 87, 88, 89], [87, 88, 89, 90], [88, 89, 90, 91], [89, 90, 91, 92], [90, 91, 92, 93], [91, 92, 93, 94], [92, 93, 94, 95], [93, 94, 95, 96], [94, 95, 96, 97], [95, 96, 97, 98], [96, 97, 98, 99]]

1行でかけるといえばかけるのだが、 for in range を使っているので少しな、という気持ちになる。とはいっても「じゃあどうすればいいのか」は少しは毎回考えているはずなのだが、結局思い浮かばず上記のコードの焼き直しを書いてしまう。

そんなに真面目に募集しているわけでもないが、もし何か思いついたかたがいれば教えて欲しい。

と締めくくろうとしたが、ちょうどいまこういう書き方を思いついた:

m = 4
a = list(range(100))
l = [a[i:i+m] for i in range(100 - m + 1)]
print(l)

スライスを使うことで内包表記のネストはなくなった。添え字アクセスをするしかないのだろうか。うーむ。

後半の添え字の + とか - とかが鬱陶しければ次のような書き方もありだろう。

m = 4
a = list(range(100))
l = [a[i:i+m] for i in range(100) if len(a[i:i+m]) == 4]
print(l)

とまぁ考えてみるとマイナーチェンジだがいろいろとでてくるものである。

*1:4つである必要はなく、10とか20とかなんでもいい