將迭代列表傳遞給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