python - 使用 Python 查找排列和组合

Question

我有 2 个变量 - a 和 b。我需要使用这些变量填充 k 个位置。所以如果 k = 3 输出应该是

[a,a,a], [a,a,b] , [a,b,a], [b,a,a], [a,b,b], [b,a,b], [b,b,a] and [b,b,b] 

输入-k

输出 - 所有组合

如何用 Python 编写代码？ itertools 在这里有什么帮助吗？

Answer 1

>>> import itertools
>>> list(itertools.product('ab', repeat=3))
[('a', 'a', 'a'), ('a', 'a', 'b'), ('a', 'b', 'a'), ('a', 'b', 'b'), ('b', 'a', 'a'), ('b', 'a', 'b'), ('b', 'b', 'a'), ('b', 'b', 'b')]