迭代Python中多个列表中的所有值组合

Posted 2021-04-08

给定多个可能变化长度的列表，我想迭代所有值的组合，每个列表中的一个项目。例如：

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

然后我希望输出为：

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

我想迭代组合列表。我怎么做到这一点？

答案

itertools.product应该做的伎俩。

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

另一答案

这可以通过使用list comprehension进行任何导入。使用你的例子：

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

combined = [(f,s) for f in first for s in second]

print(combined)
# [(1, 0.5), (1, 4), (5, 0.5), (5, 4), (8, 0.5), (8, 4)]