Python 3.3:列表的所有可能组合

时间:2014-04-07 13:23:55

标签: python list python-3.x

我有这样的清单:

[['one', 'two', 'three', ...], ['a', 'b', ...], ['left', 'right'] ...]

我需要创建所有可能的项目组合,并将其放入字符串中:

"one|a|left"
"one|a|right"
"one|b|left"
"one|b|right"
"two|a|left"
"two|a|right"
"two|b|left"
...

最简单的方法是什么?

1 个答案:

答案 0 :(得分:9)

您可以使用itertools.product

from itertools import product
lst = [['one', 'two', 'three'], ['a', 'b'], ['left', 'right']]
print(list(product(*lst)))

验证它是否符合您的要求:

[('one', 'a', 'left'), ('one', 'a', 'right'), ('one', 'b', 'left'), ('one', 'b', 'right'), ('two', 'a', 'left'), ('two', 'a', 'right'), ('two', 'b', 'left'), ('two', 'b', 'right'), ('three', 'a', 'left'), ('three', 'a', 'right'), ('three', 'b', 'left'), ('three', 'b', 'right')]

要生成您描述的所需字符串:

["|".join([p, q, r]) for p, q, r in product(*lst)]

输出:

['one|a|left',
 'one|a|right',
 'one|b|left',
 'one|b|right',
 'two|a|left',
 'two|a|right',
 'two|b|left',
 'two|b|right',
 'three|a|left',
 'three|a|right',
 'three|b|left',
 'three|b|right']