Python 中两个列表中两个字典的笛卡尔积

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}]

有什么方法可以让我只用 list/dict 理解力就能做到吗？

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}]

Answer 2

你可以试试这个。

a = [{'a': 1}, {'b': 2}]
b = [{'c': 3}, {'d': 4}]
d = [ {**i, **j} for i in a for j in b ]
print(d)

Python 中两个列表中两个字典的笛卡尔积

Cartesian product of two dict in two lists in Python

python

list-comprehension

dictionary-comprehension