遍历Python中多个列表中值的所有组合

小编典典

python

给定多个长度可能不同的列表，我想遍历值的所有组合，每个列表中的一项。例如：

first = [1, 5, 8]
second = [0.5, 4]

然后我想要的输出是：

combined = [(1, 0.5), (1, 4), (5, 0.5), (5, 4), (8, 0.5), (8, 4)]

我想遍历合并列表。我该如何完成？

阅读 241

2020-12-20

共1个答案

小编典典

itertools.product
应该可以。

>>> import itertools
>>> list(itertools.product([1, 5, 8], [0.5, 4]))
[(1, 0.5), (1, 4), (5, 0.5), (5, 4), (8, 0.5), (8, 4)]

请注意，它itertools.product返回一个迭代器，因此，如果仅要迭代一次，则无需将其转换为列表。

例如。

for x in itertools.product([1, 5, 8], [0.5, 4]):
    # do stuff

2020-12-20