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)

Я отримую KeyError, оскільки d [...] - це не список. У цьому випадку я можу додати наступний код після призначення a ініціалізації словника.

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

Чи є кращий спосіб зробити це? Скажемо, я не знаю ключів, які мені знадобляться, поки я не перейду до другого 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,]

Який найкращий спосіб впоратися з цим? В ідеалі додавання буде "просто працювати". Чи є якийсь спосіб висловити, що я хочу словник порожніх списків, навіть якщо я не знаю кожної клавіші під час першого створення списку?

Ви можете використовувати стандартний вирок :

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

Ви можете створити його з таким розумінням списку, як це:

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

А для другої частини вашого питання використовуйте вирок за замовчуванням

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

Ви можете використовувати 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).

На ваше питання вже відповіли, але IIRC ви можете замінити рядки на зразок:

if d.has_key(scope_item):

з:

if scope_item in d:

Тобто dпосилання d.keys()в тій конструкції. Іноді defaultdictце не найкращий варіант (наприклад, якщо ви хочете виконати кілька рядків коду після elseасоційованого з вищезазначеним if), і мені здається, що inсинтаксис легше читати.

Особисто я просто використовую 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'}

простий спосіб:

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

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