Что эквивалентно map <int, vector <int>> в Python?

Question

В C ++ часто делают что-то вроде этого:

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

Где я тогда использую это так:

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

Если на карте не было записи, код вставит новый пустой вектор, а затем добавит к нему.

В Python это будет выглядеть так:

myIndex = {}

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

Попытка кроме здесь немного уродлива. Есть ли способ сообщить словарю тип объекта для вставки при обнаружении ключевой ошибки во время объявления словаря?

Answer 1

Вы хотите использовать:

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

Стандартная библиотека defaultdict объекты .

Пример использования из документации 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

Что-то вроде этого возможно:

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

Answer 3

Просто чтобы завершить ответ Аластера: Существует также get эквивалент setdefault , который называется get (а не getdefault, как можно подумать):

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

Answer 4

Начиная с Python 2.5, вы можете получить поведение setdefault или использовать defaultdict, внедрив

__missing__(k)

как в примечании 10 здесь .

Answer 5

Как насчет этого? Возможно, это не оптимально для производительности, но я думаю, что это «самая простая вещь, которая могла бы работать».

myIndex = {}

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