二次元や三次元でリストを使いたいときがあると思います。しかし、自分の意図しない動きをしたという経験がある方もいらっしゃるのではないかということでpythonのリストのお話をします。
ls = [0,1,2,3,4,5,6,7,8,9]
print(ls[3])
ls.append(10)
print(ls)
一次元のリストの場合はこんな感じでしょうか?まあ違和感はないですね。次は二次元のリストを扱ってみましょう。
ls = [[0,1,2],[3,4,5],[6,7,8]]
print(ls[2][1])
ls.append([10])
ls[3].append(11)
ls[3].append(12)
print(ls)
二次元のリストはこんな感じです。まあここまではそんなに困らないと思います。
ls = [0]*10
ls_2d = [[0]*5]*2
print('ls:',ls)
print('ls_2d',ls_2d)
こんな感じでリストを作成しました。ではここでそれぞれ値を更新してみましょう。
ls[3] = 1
ls_2d[0][3] = 1;
これでlsの四番目の要素を1に、ls_2dの最初の要素の四番目を1に変更できました。
出力してみます。
print(ls)
print(ls_2d)
おかしいことに気づいたでしょうか?lsは思い通りの結果を示しています。しかし、ls_2dは最初のリストの四番目と、二番目のリストの四番目が変更されています。これは、リストの中に入っているリストがすべて同じアドレスを参照しているからです。
実際にアドレスを確認してみます。
https://docs.python.org/ja/3/reference/datamodel.html
CPython では、id(x) は x が格納されているメモリ上のアドレスを返します。
print('一つ目の要素',id(ls_2d[0]))
print('二つ目の要素',id(ls_2d[1]))
おなじですね。これを改善するにはどうすればいいのでしょうか?
いきなり回答を言ってしまうと、リスト内包表記を使います。
ls_2d_improve = [[0]*5 for i in range(2)]
ls_2d_improve[0][3] = 1
print(ls_2d_improve)
これでできました。念のためアドレスを確認しておくと、
print('一つ目の要素',id(ls_2d_improve[0]))
print('二つ目の要素',id(ls_2d_improve[1]))
違うアドレスになってますね。mutableだとかimmutableだとかそういう話はまたの機会に