¿Cuál es el equivalente de map <int, vector <int>> en Python?

Question

En C++ menudo hacer algo como esto:

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

Donde luego usarlo como esto:

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

Si no había ninguna entrada en el mapa del código insertará un nuevo vector vacío y luego anexarlo

En Python se vería así:

myIndex = {} 

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

El intento excepción es un poco feo aquí. ¿Hay alguna manera de indicarle al diccionario un tipo de objeto para insertar cuando se encuentra un KeyError en el momento de declaración del diccionario?

Answer 1

que desea utilizar:

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

biblioteca estándar defaultdict objects.

Ejemplo de uso de la documentación de Python:

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

Algo como esto quizá:

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

Answer 3

¿Qué tal esto? Puede que no sea óptimo para el rendimiento, pero creo que es "lo más simple que podría funcionar".

myIndex = {} 

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

Answer 4

sólo para completar la respuesta por Alastair: También existe la obtener equivalente de setdefault, que se llama obtener (y no getDefault, como se podría pensar):

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

Answer 5

Desde Python 2.5 y en puede obtener el comportamiento de setdefault o usar defaultdict implementando

__missing__(k) 

como en la nota 10 here.