python-3.x - 确定父列表的主要元素之间所有可能的组合

Question

我正在设计一个数据集，我正面临一个特定部分的问题。我提供了下面的示例以简化并与我的问题相关。

我有一个列表列表

list = ['b',['c','g','d'],['h','l']]

我对确定父列表主要元素之间所有可能组合的通用解决方案感兴趣

需要的解决方案：

['b','c','h']
['b','c','l']
['b','g','h']
['b','g','l']
['b','d','h']
['b','d','l']

标签： python-3.xpandasnumpy

Answer 1

您可以使用itertools.product()：

import itertools

my_list = ['b', ['c','g','d'], ['h','l']]
print(list(itertools.product(*my_list)))

输出：

[('b', 'c', 'h'), ('b', 'c', 'l'), ('b', 'g', 'h'), 
 ('b', 'g', 'l'), ('b', 'd', 'h'), ('b', 'd', 'l')]