寻找可生成指定长度双字符所有组合的Python模块函数

当然有现成的解决方案！Python的标准库itertools模块里的product()函数就是专门干这个的，完全能满足你的需求。

具体用法示例 #

假设你要生成由'a'和'b'组成的长度为3的所有字符串组合，代码可以这么写：

import itertools

# 定义可用字符集合
available_chars = ('a', 'b')
# 指定生成的字符串长度
length = 3

# 生成所有组合并转换为字符串列表
combinations = [''.join(item) for item in itertools.product(available_chars, repeat=length)]
print(combinations)

运行这段代码后，输出结果正好匹配你给出的示例：

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

原理说明 #

• itertools.product()会生成输入可迭代对象的笛卡尔积，repeat参数用来指定重复迭代输入集合的次数，也就是最终字符串的长度。
• 函数返回的是元组迭代器（比如('a','a','a')），所以我们用''.join()把每个元组拼接成完整的字符串。

更简洁的写法 #

你也可以用map()函数替代列表推导式，代码会更紧凑：

combinations = list(map(''.join, itertools.product(available_chars, repeat=length)))

而且这个方法不止局限于两个字符——哪怕你有3个、4个甚至更多字符，只要把available_chars换成对应的集合，就能生成所有可能的长度为N的字符串组合。

内容的提问来源于stack exchange，提问作者sprogissd