使用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')]

Answer 2

def genPerm(varslist, pos,resultLen, result, resultsList)
   if pos>resultLen:
       return;
   for e in varslist:
       if pos==resultLen:
           resultsList.append(result + [e]);
       else
           genPerm(varsList, pos+1, resultLen, result + [e], resultsList);

致电：

genPerm([a,b], 0, resLength, [], resultsList);