python - 如何根据给定字符和长度生成排列列表？

Question

例如，

我有一个字符列表，我想生成这些字符的排列，例如['*', '+'] 和我希望生成排列的长度，例如2.换句话说，我想找到提供的列表的所有组合，直到提供的长度。对 generatePermutations 的调用可能如下所示。

generatePermutations(['*', '+'], 2)

应该返回:

*, *
*, +
+, *
+, +

另一个例子是:

generatePermutations(['*', '+'], 3)
*, *, *
*, *, +
*, +, +
*, +, *
+, *, +
+, *, *
+, +, *
+, +, +

我该怎么做？

Answer 1

使用itertool.product。你要的是 cartesian product .

例子

>>> list(itertools.product(['*', '+'], repeat=2))
[('*', '*'), ('*', '+'), ('+', '*'), ('+', '+')]
>>> list(itertools.product(['*', '+'], repeat=3))
[('*', '*', '*'), ('*', '*', '+'), ('*', '+', '*'), ('*', '+', '+'), ('+', '*', '*'), ('+', '*', '+'), ('+', '+', '*'), ('+', '+', '+')]