【Python3】二次元配列の特定の要素に対する演算

Question

前提・実現したいこと

pythonで、入れ子構造にしたlistの特定の要素について加算をする方法を知りたい
（また、以下のやり方がうまくいかない理由を知りたい）

例) [[0, 0], [0, 0]]の3番目の0に1を足して[[0, 0], [1, 0]]としたい。

初歩的な質問で恐縮ですが、お力添えいただければ幸いです。

発生している問題・エラーメッセージ

以下のソースコードを実行すると
[[1, 0], [1, 0]]
という出力となってしまう

該当のソースコード

●Python3(Google Colaboratoryで実行、jupyterノートブック)

hoge = [[0]*2]*2
hoge[1][0] += 1
print(hoge)

Answer 1

次のようにリストを作れば良いです。

hoge = [[0 for _ in range(2)] for _ in range(2)]

Python初心者がしばしば嵌る罠であると同時に、
ちょっと文面では表現しづらいですので、次の記事の紹介を説明に代えます。

• Python のリストの扱いで注意すること - Qiita
• リストを*したときの挙動 - Qiita

---

なおこの場合リストは [[0] * 2 for _ in range(2)] で作っても良いですが...
慣れないうちはなかなか差が分かりづらいでしょう、内包表記で書く習慣を付けると良いと思います。

Comments

[nasudeng]

LouiS0616様
ご紹介頂きましたQiita記事拝読させていただきました。
識別値の仕様など、大変勉強になります。

質問した事象の原因や対策も、理解できました。
丁寧な回答、誠にありがとうございました。

Answer 2

numpyに変換してから要素を入れて、listに戻してできました(もっと簡単な方法があるかもですが...汗)

import numpy as np
hoge = [[0]*2]*2
hoge = np.array(hoge)
hoge[1][0]=1
hoge = hoge.tolist()
hoge

Comments

[LouiS0616]

NumPyを経由するのは面白いアイデアですが、
それなら hoge = np.array([[0]*2]*2).tolist(); hoge[1][0] += 1 で充分かなと思いました。

[bamboo-nova]

LouiS0616さん

いつもお世話になっております。
たしかにそれで充分ですね！ご指摘ありがとうございます。

[nasudeng]

bamboo-nova　様

NumPy使うとスッキリ書けますね...
実行速度も速いですし、実際に使う分にはこっちのがよさそうです！

回答頂きありがとうございました！！