すらぼうの開発ノート

モバイルアプリエンジニアのメモ

【Python】2次元の正方行列をリスト内包表記で生成

[[0, 1, 2, 3, 4],
 [5, 6, 7, 8, 9], 
 [10, 11, 12, 13, 14], 
 [15, 16, 17, 18, 19], 
 [20, 21, 22, 23, 24]]

上記の正方行列は次のように記述することで生成できる。

myMatrix = [[i + 5 * j for i in range(5)] for j in range(5)]

print(myMatrix)
# [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9], [10, 11, 12, 13, 14], [15, 16, 17, 18, 19], [20, 21, 22, 23, 24]]