Python：从列表列表中提取[重复]

Question

我有一个列表列表，需要提取每个项目；最终写出他们的笛卡尔积。

from itertools import product

b = {}
a = [[1, 2], [3, 4], ['a', 'b']]

for i, v in enumerate(a):
  b[i] = v

#These don't work because we are feeding a single list to product
print list(product(x for x in a)) # [([1, 2],), ([3, 4],), (['a', 'b'],)]
print list(product(b.values())) # [([1, 2],), ([3, 4],), (['a', 'b'],)]

如果我能得到这个，product 可以正常工作：

print list(product([1, 2], [3, 4], ['a', 'b']))
[(1, 3, 'a'), (1, 3, 'b'), (1, 4, 'a'), (1, 4, 'b'), (2, 3, 'a'), (2, 3, 'b'), (2, 4, 'a'), (2, 4, 'b')]

Answer 1

使用 * 将 a 解压缩到 itertools.product 中。示例 -

print list(product(*a))

解包会将a 的每个元素作为单独的参数发送到product，而不是单个列表（就像-list(product([1, 2], [3, 4], ['a', 'b']))）。

演示 -

>>> a = [[1, 2], [3, 4], ['a', 'b']]
>>> from itertools import product
>>> print(list(product(*a)))
[(1, 3, 'a'), (1, 3, 'b'), (1, 4, 'a'), (1, 4, 'b'), (2, 3, 'a'), (2, 3, 'b'), (2, 4, 'a'), (2, 4, 'b')]