Python crea un dizionario di liste

Voglio creare un dizionario i cui valori sono elenchi. Per esempio:

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

Se lo faccio:

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

Ottengo un KeyError, perché d [...] non è un elenco. In questo caso, posso aggiungere il seguente codice dopo l'assegnazione di a per inizializzare il dizionario.

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

C'è un modo migliore per farlo? Diciamo che non conosco le chiavi di cui avrò bisogno finché non sarò nel secondo forciclo. Per esempio:

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

Un'alternativa sarebbe quindi sostituire

d[scope_item].append(relation)

con

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

Qual è il modo migliore per gestirlo? Idealmente, aggiungere sarebbe "semplicemente funzionante". Esiste un modo per esprimere che desidero un dizionario di elenchi vuoti, anche se non conosco tutte le chiavi quando creo l'elenco?

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

Puoi costruirlo con la comprensione dell'elenco in questo modo:

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

E per la seconda parte della tua domanda usa 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])]

Puoi usare 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']}

La funzione piuttosto strana setdefaultdice "Ottieni il valore con questa chiave, o se quella chiave non c'è, aggiungi questo valore e poi restituiscilo."

Come altri hanno giustamente sottolineato, defaultdictè una scelta migliore e più moderna. setdefaultè ancora utile nelle versioni precedenti di Python (precedenti alla 2.5).

Alla tua domanda è già stata data una risposta, ma IIRC puoi sostituire linee come:

if d.has_key(scope_item):

con:

if scope_item in d:

Cioè, driferimenti d.keys()in quella costruzione. A volte defaultdictnon è l'opzione migliore (ad esempio, se si desidera eseguire più righe di codice dopo quelle elseassociate a quanto sopra if) e trovo che la insintassi sia più facile da leggere.

Personalmente, utilizzo JSON per convertire le cose in stringhe e viceversa. Stringhe ho capito.

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'}

il modo semplice è:

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

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