python - Python中两个列表中两个dict的笛卡尔积

Question

这是我的代码。

>>> a = [{'a': 1}, {'b': 2}]
>>> b = [{'c': 3}, {'d': 4}]

我想展示：

[{'a':1, 'c':3}, {'b':2, 'c':3}, {'a':1, 'd':4}, {'b':2, 'd':4}]

有没有办法只用列表/字典理解来做到这一点？

标签： pythonlist-comprehensiondictionary-comprehension

Answer 1

单行，没有导入解决方案可以包含一个lambda函数：

f = lambda d, c:[c] if not d else [i for k in d[0] for i in f(d[1:], {**c, **k})]

a = [{'a': 1}, {'b': 2}]
b = [{'c': 3}, {'d': 4}]
print(f([a, b], {}))

输出：

[{'a': 1, 'c': 3}, {'a': 1, 'd': 4}, {'b': 2, 'c': 3}, {'b': 2, 'd': 4}]

但是，更清洁的解决方案可以包括itertools.product：

from itertools import product
result = [{**j, **k} for j, k in product(a, b)]

输出：

[{'a': 1, 'c': 3}, {'a': 1, 'd': 4}, {'b': 2, 'c': 3}, {'b': 2, 'd': 4}]