如何获取两个列表之间的所有映射？

Question

我们有两个列表，A和B:

A = ['a','b','c']
B = [1, 2]

有没有一种Pythonic的方式来构建包含$2^n$（这里是$2^3=8$）个映射的从A到B的所有映射集？也就是说：

[(a,1), (b,1), (c,1)]
[(a,1), (b,1), (c,2)]
[(a,1), (b,2), (c,1)]
[(a,1), (b,2), (c,2)]
[(a,2), (b,1), (c,1)]
[(a,2), (b,1), (c,2)]
[(a,2), (b,2), (c,1)]
[(a,2), (b,2), (c,2)]

使用`itertools.product`，可以获取所有的元组：

import itertools as it
P = it.product(A, B)
[p for p in P]

这将会得到：

Out[3]: [('a', 1), ('a', 2), ('b', 1), ('b', 2), ('c', 1), ('c', 2)]

- Jean-Pat

Answer 1

您可以使用 itertools.product 和 zip 完成此操作。

from itertools import product
print [zip(A, item) for item in product(B, repeat=len(A))]

输出

[[('a', 1), ('b', 1), ('c', 1)],
 [('a', 1), ('b', 1), ('c', 2)],
 [('a', 1), ('b', 2), ('c', 1)],
 [('a', 1), ('b', 2), ('c', 2)],
 [('a', 2), ('b', 1), ('c', 1)],
 [('a', 2), ('b', 1), ('c', 2)],
 [('a', 2), ('b', 2), ('c', 1)],
 [('a', 2), ('b', 2), ('c', 2)]]

product(B, repeat=len(A))会产生：

[(1, 1, 1),
 (1, 1, 2),
 (1, 2, 1),
 (1, 2, 2),
 (2, 1, 1),
 (2, 1, 2),
 (2, 2, 1),
 (2, 2, 2)]

然后我们从乘积中选取每个元素，并将其与A进行压缩，以得到你想要的输出结果。

- thefourtheye

Answer 2

import itertools as it

A = ['a','b','c']
B = [1, 2]

for i in it.product(*([B]*len(A))):
    print(list(zip(A, i)))

输出:

[('a', 1), ('b', 1), ('c', 1)]
[('a', 1), ('b', 1), ('c', 2)]
[('a', 1), ('b', 2), ('c', 1)]
[('a', 1), ('b', 2), ('c', 2)]
[('a', 2), ('b', 1), ('c', 1)]
[('a', 2), ('b', 1), ('c', 2)]
[('a', 2), ('b', 2), ('c', 1)]
[('a', 2), ('b', 2), ('c', 2)]

不确定它是否非常符合Python风格，但如果您看一下 it.product(*([B]*len(A)))，它就是，因为它使用了多个特定于Python的语言特性。但实际上它太神秘了，不太符合Python风格... B基于A的长度被重复n次并解包到product函数中。

- CodeManX