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

Question

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

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

然后我希望输出为：

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

我想遍历组合列表。我该怎么做？

Answer 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

Answer 2

这可以在没有任何导入的情况下使用列表理解来实现。使用您的示例：

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