将迭代列表传递给 itertools 函数

Question

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

说清楚，我想要

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

但从这样的嵌套列表输入生成

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