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')]

Python : 从列表列表中提取

Python : extract from list of lists

python

list

itertools