Объединение списков значений словаря в python
Я пытаюсь объединить три словаря, все из которых имеют одинаковые ключи, а также списки значений или одиночные значения.
one={'a': [1, 2], 'c': [5, 6], 'b': [3, 4]}
two={'a': [2.4, 3.4], 'c': [5.6, 7.6], 'b': [3.5, 4.5]}
three={'a': 1.2, 'c': 3.4, 'b': 2.3}
Мне нужно, чтобы все элементы в значениях добавлялись в один список.
result={'a': [1, 2, 2.4, 3.4, 1.2], 'c': [5, 6, 5.6, 7.6, 2.3], 'b': [3, 4, 3.5, 4.5, 3.4]}
Я пробовал несколько вещей, но большинство помещал значения во вложенные списки.
Например.
out=dict((k, [one[k], two.get(k), three.get(k)]) for k in one)
{'a': [[1, 2], [2.4, 3.4], 1.2], 'c': [[5, 6], [5.6, 7.6], 3.4], 'b': [[3, 4], [3.5, 4.5], 2.3]}
Я попробовал обновить его, выполнив следующие значения:
out.update((k, [x for x in v]) for k,v in out.iteritems())
но результаты были точно такими же.
Я попытался просто добавить списки, но поскольку третий словарь имеет только float, я не мог этого сделать.
check=dict((k, [one[k]+two[k]+three[k]]) for k in one)
Итак, я попытался сначала добавить списки в значениях один и два, а затем добавить значение три. Добавление списков сработало хорошо, но затем, когда я попытался добавить float из третьего словаря, внезапно все значение переместилось в "None"
check=dict((k, [one[k]+two[k]]) for k in one)
{'a': [[1, 2, 2.4, 3.4]], 'c': [[5, 6, 5.6, 7.6]], 'b': [[3, 4, 3.5, 4.5]]}
new=dict((k, v.append(three[k])) for k,v in check.items())
{'a': None, 'c': None, 'b': None}
Ответы
Ответ 1
Как однострочный, со словарным пониманием:
new = {key: value + two[key] + [three[key]] for key, value in one.iteritems()}
Это создает новые списки, объединяя список из one
с соответствующим списком из two
, помещая одно значение в three
во временный список, чтобы упростить конкатенацию.
Или с обновлением цикла for
one
на месте:
for key, value in one.iteritems():
value.extend(two[key])
value.append(three[key])
Это использует list.extend()
для обновления исходного списка на месте с помощью списка из two
и list.append()
для добавления единственного значения из three
.
Если вы поступили не так:
-
ваша первая попытка создает новый список со значениями из one
, two
и three
, вложенными внутри, а не конкатенацией существующих списков. Ваша попытка очистить это просто скопировала эти вложенные списки.
-
Ваша вторая попытка не сработала, потому что значение в three
не является списком, поэтому не может быть объединено. Я создал новый список только для этого значения.
-
Ваша последняя попытка не должна использоваться list.append()
в выражении генератора, потому что вы сохраняете возвращаемое значение этого метода, который всегда None
(его изменение хранится в v
напрямую, а список не нуждается в возврате снова).
Демонстрация первого подхода:
>>> one={'a': [1, 2], 'c': [5, 6], 'b': [3, 4]}
>>> two={'a': [2.4, 3.4], 'c': [5.6, 7.6], 'b': [3.5, 4.5]}
>>> three={'a': 1.2, 'c': 3.4, 'b': 2.3}
>>> {key: value + two[key] + [three[key]] for key, value in one.iteritems()}
{'a': [1, 2, 2.4, 3.4, 1.2], 'c': [5, 6, 5.6, 7.6, 3.4], 'b': [3, 4, 3.5, 4.5, 2.3]}
Ответ 2
Произвольный словарь, номер и ключи
Проблемы с вашей попыткой покрываются решением @MartijnPieters.
Для обобщенного решения рассмотрите возможность использования itertools.chain
для itertools.chain
нескольких словарей. Вы также можете использовать defaultdict
для более общего случая, когда вы не найдете одинаковые ключи в каждом словаре.
from collections import defaultdict
from itertools import chain
from operator import methodcaller
# dictionaries with non-equal keys, values all lists for simplicity
one = {'a': [1, 2], 'c': [5, 6], 'b': [3, 4], 'e': [6.2]}
two = {'a': [2.4, 3.4], 'c': [5.6, 7.6], 'b': [3.5, 4.5], 'f': [1.3]}
three = {'a': [1.2], 'c': [3.4], 'b': [2.3], 'e': [3.1]}
# initialise defaultdict of lists
dd = defaultdict(list)
# iterate dictionary items
dict_items = map(methodcaller('items'), (one, two, three))
for k, v in chain.from_iterable(dict_items):
dd[k].extend(v)
print(dd)
# defaultdict(list,
# {'a': [1, 2, 2.4, 3.4, 1.2],
# 'b': [3, 4, 3.5, 4.5, 2.3],
# 'c': [5, 6, 5.6, 7.6, 3.4],
# 'e': [6.2, 3.1],
# 'f': [1.3]})
Обратите внимание, что defaultdict
является подклассом dict
поэтому обычно нет необходимости преобразовывать результат в обычный dict
.
Ответ 3
См. эту справку или нет:
>>> dic={}
>>> k=[]
>>> for i in 'abc':
k=one[i]+two[i]
k.append(three[i])
dic[i]=k
>>> dic
{'c': [5, 6, 5.6, 7.6, 3.4], 'a': [1, 2, 2.4, 3.4, 1.2], 'b': [3, 4, 3.5, 4.5, 2.3]}
Ответ 4
Надежное решение. знак равно
def FullMergeDict(D1, D2):
for key, value in D1.items():
if key in D2:
if type(value) is dict:
FullMergeDict(D1[key], D2[key])
else:
if type(value) in (int, float, str):
D1[key] = [value]
if type(D2[key]) is list:
D1[key].extend(D2[key])
else:
D1[key].append(D2[key])
for key, value in D2.items():
if key not in D1:
D1[key] = value
if __name__ == '__main__':
X = {
'a': 'aaa',
'c': [1,3,5,7],
'd': 100,
'e': {'k': 1, 'p': 'aa','t': [-1,-2]},
'f': {'j':1}
}
Y = {
'b': 'bbb',
'd': 200,
'e': {'k': 2, 'p': 'bb','o': [-4]},
'c': [2,4,6],
'g': {'v':2}
}
FullMergeDict(X, Y)
exit(0)
Результат:
X = {
'a': 'aaa',
'b': 'bbb',
'c': [1, 3, 5, 7, 2, 4, 6],
'd': [100, 200],
'e': {'k': [1, 2], 'o': [-4], 'p': ['aa', 'bb'], 't': [-1, -2]},
'f': {'j': 1},
'g': {'v': 2}}