python - 将可迭代列表传递给 itertools 函数

Question

我正在使用 itertools.product 功能。我有一个 2 深的嵌套列表，它是一个可迭代的列表。我想将此传递给产品功能，不知道如何正确格式化。

说清楚，我想要

In [37]: [k for k in product([1,2],['a','b'])]
Out[37]: [(1, 'a'), (1, 'b'), (2, 'a'), (2, 'b')]

但是从这样的nested_list输入生成

nested_list = [[1,2],['a','b']]

但相反我得到

In [36]: [k for k in product(nested_list)]
Out[36]: [([1, 2],), (['a', 'b'],)]

Answer 1

product接受可变数量的参数，因此您需要解压缩列表。

list(product(*nested_list)) # without list() normally, of course