Как объединить два списка словарей без потери словаря с несовпадающим ключем на Python 3?

Question

[Ivan_R2]

Здравствуйте!
Есть два списка словарей:

a = [{'code': 'a1', 'total1': 7}, {'code': 'a2', 'total1': 17}]
b = [{'code': 'a1', 'total2': 55}, {'code': 'a3', 'total2': 22}]

Как на выходе получить:
1)

l = [{'code': 'a1', 'total1': 7, 'total2': 55}, 
      {'code': 'a2', 'total1': 17}, 
      {'code': 'a3', 'total2': 22}]

2)

l = [{'code': 'a1', 'total1': 7, 'total2': 55}, 
      {'code': 'a2', 'total1': 17, 'total2': 0}, 
      {'code': 'a3', 'total2': 22, 'total1':0}]

Пробовал такие варианты:

l = list()
for i, j in zip(a, b):  # Проходим по элементам списков
    i.update(j)  # Записываем всё в один словарь
    l.append(i)

и l = [dict(i, **j) for i, j in zip(a, b)]
Но результатом был список

[{'code': 'a1', 'total1': 7, 'total2': 55}, {'code': 'a3', 'total1': 17, 'total2': 22}]

Желательно, сделать так, чтобы функция не была слишком медлительной.
Спасибо

Answer 1

[Roman K]

from itertools import chain

a = [{'code': 'a1', 'total1': 7}, {'code': 'a2', 'total1': 17}]
b = [{'code': 'a1', 'total2': 55}, {'code': 'a3', 'total2': 22}]

def f(a, b):
    cache = {}
    for dct in chain(a, b):
        code = dct["code"]
        cache.setdefault(code, dct).update(dct)

    return list(cache.values())

f(a, b)

Нули расставить по вкусу. Алгоритм ест дополнительную память, но по скорости вряд ли можно что-то быстрей.

Comments

[Ivan_R2]

Огромное спасибо!

Answer 2

[aRegius]

import itertools
from collections import defaultdict

temp_data = defaultdict(dict)

for item in itertools.chain(a, b):
      temp_data[item['code']].update(item)

data = list(temp_data.values())

Comments

[Roman K]

Примерно на 27% медленней моего варианта.

[Ivan_R2]

А для варианта 2) только потом циклом проходится или можно сразу как-то сделать?

[aRegius]

Ivan_R2, циклом

[Ivan_R2]

aRegius, спасибо