python - 将嵌套列表简化为笛卡尔积

Question

我在 Python 中有一个嵌套列表:

lst = ['alpha', ['beta', 'gamma'], ['delta', 'peta', 'lambda']]

我需要一个函数来返回一个列表，其中包含它们的笛卡尔积。好吧，我认为笛卡尔积不是一个正确的词，但仍然以合乎逻辑的方式结果看起来像:

final_lst = your_magical_function(lst)
print final_lst

'''
[['alpha','beta','delta'],
['alpha','beta','peta'],
['alpha','beta','lambda'],
['alpha','gamma','delta'],
['alpha','gamma','peta']
['alpha','gamma','lambda']]
'''

不管有没有递归的函数，都欢迎。

Answer 1

使用itertools.product ，这将要求您稍微修改输入('alpha' 为 ['alpha']):

from itertools import product

lst = [['alpha'],['beta','gamma'],['delta','peta','lambda']]

for res in product(*lst):
    print(res)

>> ('alpha', 'beta', 'delta')

('alpha', 'beta', 'peta')
('alpha', 'beta', 'lambda')
('alpha', 'gamma', 'delta')
('alpha', 'gamma', 'peta')
('alpha', 'gamma', 'lambda')