Преобразование списка кортежей в dict

У меня есть список таких кортежей:

[
('a', 1),
('a', 2),
('a', 3),
('b', 1),
('b', 2),
('c', 1),
]

Я хочу повторить этот ключ по первому элементу, поэтому, например, я мог бы напечатать что-то вроде этого:

a 1 2 3
b 1 2
c 1

Как мне это сделать, не сохраняя элемент, чтобы отслеживать, является ли первый элемент таким же, как я зацикливаюсь на кортежах? Это кажется довольно грязным (плюс мне нужно отсортировать список для начала)...


python list dictionary tuples iteration
person Dan    schedule 04.11.2008    source источник

Ответы (5)


arrow_upward
42
arrow_downward 

l = [
('a', 1),
('a', 2),
('a', 3),
('b', 1),
('b', 2),
('c', 1),
]

d = {}
for x, y in l:
    d.setdefault(x, []).append(y)
print d

производит:

{'a': [1, 2, 3], 'c': [1], 'b': [1, 2]}
person Ned Batchelder    schedule 04.11.2008

arrow_upward
37
arrow_downward

Чуть проще...

>>> from collections import defaultdict
>>> fq= defaultdict( list )
>>> for n,v in myList:
        fq[n].append(v)

>>> fq
defaultdict(<type 'list'>, {'a': [1, 2, 3], 'c': [1], 'b': [1, 2]})
person S.Lott    schedule 04.11.2008
comment
defaultdict определенно подходит для этого. На самом деле, если вы посмотрите на пример в документации, он очень похож на эту проблему — docs.python.org/library/collections.html#defaultdict-examples - person Aaron Newton; 02.06.2012

arrow_upward
9
arrow_downward

Решение с использованием groupby

    >>> from itertools import groupby
    >>> l = [('a',1), ('a', 2),('a', 3),('b', 1),('b', 2),('c', 1),]
    >>> [(label, [v for l,v in value]) for (label, value) in groupby(l, lambda x:x[0])]
    [('a', [1, 2, 3]), ('b', [1, 2]), ('c', [1])]

groupby(l, lambda x:x[0]) дает вам итератор, который содержит ['a', [('a', 1), ...], c, [('c', 1)], . ..]

person user7461    schedule 04.11.2008
comment
Вероятно, вам следует упомянуть, что для этого вам нужен отсортированный список (или используйте groupby(sorted(l),..), чтобы это работало. Приведенные данные выглядят отсортированными, но в вопросе упоминается (плюс мне нужно отсортировать список, чтобы начать с)... так что я не уверен, что вы можете положиться на это. - person Brian; 04.11.2008

arrow_upward
3
arrow_downward

я бы просто сделал основное

answer = {}
for key, value in list_of_tuples:
  if key in answer:
    answer[key].append(value)
  else:
    answer[key] = [value]

Если это так коротко, зачем использовать что-то сложное. Конечно, если вы не возражаете против использования setdefault, это тоже нормально.

person lacker    schedule 04.11.2008

arrow_upward
2
arrow_downward

Вывести список кортежей, сгруппированных по первому элементу

Этот ответ основан на @gommen один.

#!/usr/bin/env python

from itertools import groupby
from operator  import itemgetter

L = [
('a', 1),
('a', 2),
('a', 3),
('b', 1),
('b', 2),
('c', 1),
]

key = itemgetter(0)
L.sort(key=key) #NOTE: use `L.sort()` if you'd like second items to be sorted too
for k, group in groupby(L, key=key):
    print k, ' '.join(str(item[1]) for item in group)

Выход:

a 1 2 3
b 1 2
c 1
person jfs    schedule 07.11.2009