python(3.7)の配列に要素を追加する

Question

前提・実現したいこと

pythonで辞書型を配列に追加していきたいです．その中で辞書型の["test1"]の部分だけ連番を振りたいです．以下の問題につてご教授ください．

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

[{'test1': 0, 'test2': 2, 'test3': 3, 'test4': 4, 'test5': 5},
 {'test1': 1, 'test2': 2, 'test3': 3, 'test4': 4, 'test5': 5},
 {'test1': 2, 'test2': 2, 'test3': 3, 'test4': 4, 'test5': 5},
 {'test1': 3, 'test2': 2, 'test3': 3, 'test4': 4, 'test5': 5},
 {'test1': 4, 'test2': 2, 'test3': 3, 'test4': 4, 'test5': 5}]

上記のような辞書型を格納した配列を作成したいのですが以下の該当のソースコードを実行すると以下の配列が生成されます．

[{'test1': 4, 'test2': 2, 'test3': 3, 'test4': 4, 'test5': 5},
 {'test1': 4, 'test2': 2, 'test3': 3, 'test4': 4, 'test5': 5},
 {'test1': 4, 'test2': 2, 'test3': 3, 'test4': 4, 'test5': 5},
 {'test1': 4, 'test2': 2, 'test3': 3, 'test4': 4, 'test5': 5},
 {'test1': 4, 'test2': 2, 'test3': 3, 'test4': 4, 'test5': 5}]

辞書型の["test1"]が変動せずにrange(5)の最後の値である4に固定されたものが作成されます．

該当のソースコード

agent = {}
agent_list = []
for i in range(5):
    agent["test1"] = i
    agent["test2"] = 2
    agent["test3"] = 3
    agent["test4"] = 4
    agent["test5"] = 5
    agent_list.append(agent)

print(agent_list)

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Answer 1

同じオブジェクトを使いまわしているのが問題です。
次のようにループ内でagentを生成すれば良いかと。

agent_list = []

for i in range(5):
    agent = {}
    agent["test1"] = i
    agent["test2"] = 2
    agent["test3"] = 3
    agent["test4"] = 4
    agent["test5"] = 5
    agent_list.append(agent)

print(agent_list)

辞書リテラルを使った方が簡単なのですが。

agent_list = []

for i in range(5):
    agent_list.append({
        'test1': i, 'test2': 2, 'test3': 3, 'test4': 4, 'test5': 5
    })

print(agent_list)

Comments

[kurata-yuhei]

ありがとうございます．提示いただいた方法は知りませんでした．試してみます．

Answer 2

agent = {}  # ここで辞書オブジェクトのインスタンス生成
agent_list = []
for i in range(5):
    agent["test1"] = i  # 以下全て同じ辞書オブジェクトである
    agent["test2"] = 2
    agent["test3"] = 3
    agent["test4"] = 4
    agent["test5"] = 5
    agent_list.append(agent)

print(agent_list)


# わかりやすくするために勝手に追加
from pprint import pprint
agent_list[0]["test1"] = "hoge"
pprint(agent_list)
""" =>
[{'test1': 'hoge', 'test2': 2, 'test3': 3, 'test4': 4, 'test5': 5},
 {'test1': 'hoge', 'test2': 2, 'test3': 3, 'test4': 4, 'test5': 5},
 {'test1': 'hoge', 'test2': 2, 'test3': 3, 'test4': 4, 'test5': 5},
 {'test1': 'hoge', 'test2': 2, 'test3': 3, 'test4': 4, 'test5': 5},
 {'test1': 'hoge', 'test2': 2, 'test3': 3, 'test4': 4, 'test5': 5}]
"""

対策：ループ内でリストの要素に入れる各辞書ごとに新しいインスタンスを生成する必要があります。

agent_list = []
for i in range(5):
    agent = {}  # ここで作るべき
    agent["test1"] = i 
    agent["test2"] = 2
    agent["test3"] = 3
    agent["test4"] = 4
    agent["test5"] = 5
    agent_list.append(agent)

print(agent_list)

Comments

[kurata-yuhei]

ありがとうございます．インスタンスをその都度生成していなかったため同じインスタンスが使われて最後で更新されていたのですね．解決いたしました．

Answer 3

すでに質問文のコードを修正する方法が回答されていますが、他の書き方をいくつか示します。
ポイントは、

• dict で "test2": 2 , ... "test5": 5} の変化しない部分を other_dict という変数にしてそれを利用する方法。
• other_dict 部分を内包表記で生成する方法。

dict の連結、内包表記、map の利用などをいろいろ試してみてください。

ag.py

other_agent = {
    "test2": 2,
    "test3": 3,
    "test4": 4,
    "test5": 5
}

agent_list = []
for i in range(5):
    agent = {"test1": i}
    agent.update(other_agent)
    agent_list.append(agent)
print(agent_list)
print()

agent_list = []
for i in range(5):
    first_item = {"test1": i}
    agent_list.append({**first_item, **other_agent})
print(agent_list)
print()

agent_list = [{**{"test1": i}, **other_agent} for i in range(5)]
print(agent_list)
print()

agent_list = list(map(lambda x: dict({"test1": x}, **other_agent), range(5)))
print(agent_list)
print()

agent_list = []
for i in range(5):
    agent = {"test" + str(x): x if x != 1 else i for x in range(1, 6)}
    agent_list.append(agent)
print(agent_list)
print()

agent_list = [{"test" + str(x): x if x != 1 else i for x in range(1, 6)} for i in range(5)]
print(agent_list)
print()

実行例