列表元素列表的组合

Question

我想要一个列表，它是列表元素列表的组合例如：我的输入

x = [['P'], ['E', 'C'], ['E', 'P', 'C']]

输出应该是

['PEE','PEP','PEC','PCE','PCP','PCC']]

非常感谢任何帮助。

Answer 1

使用itertools

[''.join(i) for i in itertools.product(*x)]

注意：假设最后一个应该是'PCC'

Answer 2

这里有一个解决方案

def comb(character_list_list):
  res = ['']
  for character_list in character_list_list:
    res = [s+c for s in res for c in character_list]
  return res

在你的例子中，它给出了预期的结果

>>> comb([['P'], ['E', 'C'], ['E', 'P', 'C']])
['PEE', 'PEP', 'PEC', 'PCE', 'PCP', 'PCC']

使用 functools.reduce() 可以实现更短的版本，但不推荐使用此函数。

列表元素列表的组合

Combination of list of list elements

python

combinatorics

python-3.x