找到N长度的两个字符的所有可能组合

Question

我有两个字符，例如：

a = 'a'
b = 'b'

我需要找到这两个字符的所有可能组合，这些字符将构成一个长度为N的字符串。例如，如果N = 3：

combinations = ['aab', 'aba', 'baa', 'abb', 'bab', 'bba', 'aaa', 'bbb']

某些模块中是否有功能？

Answer 1

itertools.product就是你想要的：

>>> from itertools import product
>>> a = 'a'
>>> b = 'b'
>>> N = 3
>>> lst = [a, b]
>>> [''.join(x) for x in product(lst, repeat = N)]
['aaa', 'aab', 'aba', 'abb', 'baa', 'bab', 'bba', 'bbb']

也可以用三重嵌套列表理解来编写：

>>> [x + y + z for x in lst for y in lst for z in lst]
['aaa', 'aab', 'aba', 'abb', 'baa', 'bab', 'bba', 'bbb']