Преобразование списка кортежей в Dict в Python
-
06-07-2019 - |
Вопрос
У меня есть список таких кортежей:
[
('a', 1),
('a', 2),
('a', 3),
('b', 1),
('b', 2),
('c', 1),
]
Я хочу перебрать этот ключ по первому элементу, поэтому, например, я мог бы напечатать что-то вроде этого:
a 1 2 3
b 1 2
c 1
Как мне это сделать, не сохраняя элемент для отслеживания того, является ли первый элемент таким же, как я делаю цикл по кортежам?Это выглядит довольно запутанно (плюс для начала мне нужно отсортировать список)...
Решение
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]}
Другие советы
Чуть проще ...
>>> 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]})
Решение с использованием 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) ], ...]
Распечатать список кортежей, сгруппированных по первому элементу
Этот ответ основан на тот @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
Я бы просто сделал основной
answer = {} for key, value in list_of_tuples: if key in answer: answer[key].append(value) else: answer[key] = [value]
Если это так коротко, зачем использовать что-то сложное. Конечно, если вы не против использовать setdefault, это тоже нормально.