web-dev-qa-db-ja.com

Pythonはリストの辞書を作成する

値がリストである辞書を作りたいです。例えば:

{
  1: ['1'],
  2: ['1','2'],
  3: ['2']
}

私が行った場合:

d = dict()
a = ['1', '2']
for i in a:
    for j in range(int(i), int(i) + 2): 
        d[j].append(i)

D [...]はリストではないので、KeyErrorが出ます。この場合、aの割り当ての後に次のコードを追加して辞書を初期化できます。

for x in range(1, 4):
    d[x] = list()

これを行うより良い方法はありますか? 2番目のforループに入るまで、必要なキーがわからないとしましょう。例えば:

class relation:
    scope_list = list()
...
d = dict()
for relation in relation_list:
    for scope_item in relation.scope_list:
        d[scope_item].append(relation)

代替手段はそれから取り替えである

d[scope_item].append(relation)

if d.has_key(scope_item):
    d[scope_item].append(relation)
else:
    d[scope_item] = [relation,]

これを処理するための最良の方法は何ですか?理想的には、追記は「うまくいく」でしょう。最初にリストを作成したときにすべてのキーがわからない場合でも、空のリストの辞書が必要であることを表現する方法はありますか。

203
user118662

あなたは defaultdict を使うことができます。

>>> from collections import defaultdict
>>> d = defaultdict(list)
>>> for i in a:
...   for j in range(int(i), int(i) + 2):
...     d[j].append(i)
...
>>> d
defaultdict(<type 'list'>, {1: ['1'], 2: ['1', '2'], 3: ['2']})
>>> d.items()
[(1, ['1']), (2, ['1', '2']), (3, ['2'])]
263
bernie

あなたはこのようなリスト内包表記でそれを構築することができます:

>>> dict((i, range(int(i), int(i) + 2)) for i in ['1', '2'])
{'1': [1, 2], '2': [2, 3]}

そしてあなたの質問の2番目の部分には defaultdict を使ってください。

>>> from collections import defaultdict
>>> s = [('yellow', 1), ('blue', 2), ('yellow', 3), ('blue', 4), ('red', 1)]
>>> d = defaultdict(list)
>>> for k, v in s:
        d[k].append(v)

>>> d.items()
[('blue', [2, 4]), ('red', [1]), ('yellow', [1, 3])]
48
Nadia Alramli

setdefaultを使う:

d = dict()
a = ['1', '2']
for i in a:
    for j in range(int(i), int(i) + 2): 
        d.setdefault(j, []).append(i)

print d  # prints {1: ['1'], 2: ['1', '2'], 3: ['2']}

やや奇妙な名前のsetdefault関数は、「このキーを使って値を取得するか、そのキーが存在しない場合はこの値を追加してから返します」と言っています。

編集:他の人が正しく指摘しているように、defaultdictはより良い、より現代的な選択です。 setdefaultは古いバージョンのPython(2.5より前)でもまだ便利です。

29
RichieHindle

あなたの質問はすでに答えられています、しかしIIRCあなたは次のような行を置き換えることができます:

if d.has_key(scope_item):

と:

if scope_item in d:

つまり、その構造の中でdd.keys()を参照しています。 defaultdictが最善の選択肢ではない場合もあります(例えば、上記のelseに関連付けられたifの後に複数行のコードを実行したい場合)、そしてin構文が読みやすくなります。

2
spiffyman

個人的には、JSONを使って文字列に変換したり戻したりします。私が理解している文字列.

import json
s = [('yellow', 1), ('blue', 2), ('yellow', 3), ('blue', 4), ('red', 1)]
mydict = {}
hash = json.dumps(s)
mydict[hash] = "whatever"
print mydict
#{'[["yellow", 1], ["blue", 2], ["yellow", 3], ["blue", 4], ["red", 1]]': 'whatever'}
1
john ktejik

簡単な方法は次のとおりです。

a = [1,2]
d = {}
for i in a:
  d[i]=[i, ]

print(d)
{'1': [1, ], '2':[2, ]}
1
Henning Lee