列表元素列表的组合

时间:2018-05-24 13:54:35

标签: python python-3.x combinatorics

我想要一个列表,它是列表元素列表的组合 例如: 我的意见

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

输出应为

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

非常感谢任何帮助。

2 个答案:

答案 0 :(得分:4)

使用itertools

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

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

答案 1 :(得分:1)

这是一个解决方案

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()可以使用更短的版本,但不推荐使用此功能。