Was ist äquivalent zu map <int, vector <int>> in Python?

Question

In C++ tun oft so etwas wie diese:

typedef map<int, vector<int> > MyIndexType; 

Wo ich es dann wie folgt verwenden:

MyIndexType myIndex; 
for(... some loop ...) 
{ 
    myIndex[someId].push_back(someVal); 
} 

Wenn in der Karte gibt es keinen Eintrag den Code ein neue leere Vektor einfügen und dann anhängen.

In Python würde es so aussehen:

myIndex = {} 

for (someId,someVal) in collection: 
    try: 
     myIndex[someId].append(someVal) 
    except KeyError: 
     myIndex[someId] = [someVal] 

Der Versuch, außer ein bisschen hier hässlich. Gibt es eine Möglichkeit, dem Wörterbuch einen Objekttyp mitzuteilen, der eingefügt werden soll, wenn ein KeyError bei der Deklarationszeit des Wörterbuchs auftritt?

Answer 1

Sie verwenden möchten:

from collections import defaultdict 
myIndex = defaultdict(list) 
myIndex[someId].append(someVal) 

Standard Library defaultdict objects.

Beispiel für die Verwendung von der Python-Dokumentation:

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

Answer 2

So etwas wie dies vielleicht:

myIndex = {} 
for (someId,someVal) in collection: 
    myIndex.setdefault(someId, []).append(someVal) 

Answer 3

Wie wäre das? Es ist möglicherweise nicht leistungsoptimal, aber ich denke, es ist die "einfachste Sache, die möglicherweise funktionieren könnte".

myIndex = {} 

for (someId,someVal) in collection: 
    if someId not in myIndex: 
     myIndex[someId] = [] 
    myIndex[someId].append(someVal) 

Answer 4

nur die Antwort von Alastair abzuschließen: Es gibt auch die erhalten Äquivalent setdefault, die aufgerufen wird, erhalten (und nicht getDefault, wie man denken könnte):

myIndex = {} 
someId = None 
myList = myIndex.get(someId, []) # myList is [] now 

Answer 5

Von Python 2.5 und können Sie das Verhalten von setdefault oder defaultdict durch die Implementierung

erhalten

wie in Anmerkung 10 here.