列表元素列表的组合[重复]

Posted 2023-02-24

【中文标题】列表元素列表的组合[重复]

【英文标题】：Combination of list of list elements [duplicate]

【发布时间】：2018-11-03 18:56:10

【问题描述】：

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

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

输出应该是

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

非常感谢任何帮助。

【问题讨论】：

最后一个不应该是PCC吗？

【参考方案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() 可以实现更短的版本，但不建议使用此函数。

【参考方案2】：

使用itertools

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

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