python - 我可以在 itertools 列表中添加一个变量名吗？

Question

这将打印所有可能的组合:

import itertools
iterables = [[0,1,2], ["3","46","5"]]
for item in itertools.product(*iterables, repeat=1):
    print(item)

喜欢:

0,3
0,46
0,5
1,3
etc.

假设“A”具有值 [0,1,2]而“B”有 ["3", "46", "5] ,
我可以修改打印将是这样的代码:

A 0
B 3
A 0
B 46
A 1
B 3

等等。？
或者变量名也是结果的一部分的任何其他方式？
非常感谢。

Answer 1

展平元组列表，并使用重复的显式变量名称序列压缩它。

from itertools import chain, cycle, product

for var, value in zip(cycle(["A", "B"]),
                      chain.from_iterable(product(*iterables))):
    print(var, value)

chain.from_iterable变成类似 (x, y), (x, y)进入 x, y, x, y . cycle([x,y])生产 x, y, x, y, ... .循环是无限的，但 zip仅消耗所需的量以耗尽扁平化的产品序列。