Python - itertools.product() 内的 for 循环

Question

这个问题已经有答案了:

Python itertools.product with arbitrary number of sets (2 个回答)

已关闭 4 年前。

我现在脑子有问题。

我有大量变量，每个变量可以处于两种状态。我想将它们组合成一个数组，为此我选择了 itertools.product。

所以基本上我需要的是类似的东西

a = ["A", "a"]
b = ["B", "b"]
c = ["C", "c"]
...
result = list(itertools.product(a,b,c,...)

如果我手写a、b、c，效果很好，但写那么多会很烦人。到目前为止，我的方法是将我的状态存储在我循环更新的字典中:

dict = {1: ["A","a"], 2: ["B","b"], ..., 300: ["XYZ", "xyz"]}

现在我想合并它

result = list(itertools.product([dict[i] for i in range(1,4)]))

现在我期待类似的事情

("A","B","C"),("A","B","c"),("A","b","C"),...,("a","b","c")

但是我得到了

(['A', 'a'],), (['B', 'b'],), (['C', 'c'],)

这是怎么发生的？又该如何解决呢？

Answer 1

itertools.product 接受可变数量的参数，而不是参数列表。因此，只需在列表前面放置一个 * 即可打开列表

itertools.product(*[dict[i] for i in range(1,4)])