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


214

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

{
  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,]

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

回答:


278

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

>>> from collections import defaultdict
>>> d = defaultdict(list)
>>> a = ['1', '2']
>>> 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'])]

1
collectionsモジュールの下の他の辞書もこのように機能しますcollections.OrderedDict
txsaw1

2
ああ。これは素晴らしい。また、「= []」に初期化する必要はありません。いい物!
Wilmer E. Henao

1
NameError: name 'a' is not defined
Sアンドリュー

51

次のようなリスト内包表記を使用して作成できます。

>>> 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])]

32

使用できます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より前)でも引き続き役立ちます。


2
これは機能しますが、通常はdefaultdictが使用可能な場合にそれを使用することをお勧めします。
デビッドZ

@David、ええ、setdefaultはデザインの最も素晴らしいビットではありませんでした。申し訳ありません。これは決して最良の選択ではありません。私たち(Pythonコミッター)は、collections.defaultdictで集団評判を取り戻したと思いますが、;-)。
Alex Martelli、

@ DavidZ、setdefaultはdefaultdictとは異なり、柔軟性が高くなります。それ以外の場合は、異なる辞書キーに異なるデフォルト値をどのように指定しますか?
Alex Gidan

@AlexGidanそのとおりですが、この質問には特に関係ありません。
David Z

この回答は、OrderedDictとデフォルト値が必要な場合にも役立ちます。
nimcap

2

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

if d.has_key(scope_item):

と:

if scope_item in d:

つまり、その構造でのd参照d.keys()です。defaultdict(たとえば、else上記に関連付けられた後に複数行のコードを実行する場合など)は最良のオプションではない場合があり、構文が読みやすいifと思いますin


2

個人的には、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'}

弊社のサイトを使用することにより、あなたは弊社のクッキーポリシーおよびプライバシーポリシーを読み、理解したものとみなされます。
Licensed under cc by-sa 3.0 with attribution required.