Python 2.7中如何拼接itertools.product()结果并生成多长度字符组合？

嘿，我来帮你搞定这个Python 2.7里的字符组合生成问题！

先解决你的示例需求：生成A-C、长度1和2的所有组合 #

首先，itertools.product确实是干这个活的绝佳工具——它能帮你生成字符集的笛卡尔积，刚好对应所有可能的组合。不过它默认只生成固定长度的组合，所以要覆盖多个长度，我们只需要循环不同的长度值，再把结果合并就行。

直接上能跑的代码：

import itertools

# 定义你的字符子集
chars = ['A', 'B', 'C']
# 要生成的组合长度
target_lengths = [1, 2]

# 用列表推导式一次性生成所有组合
all_combinations = [''.join(combination) for length in target_lengths 
                    for combination in itertools.product(chars, repeat=length)]

print(all_combinations)

运行后你会得到预期的结果：

['A', 'B', 'C', 'AA', 'AB', 'AC', 'BA', 'BB', 'BC', 'CA', 'CB', 'CC']

说说你遇到的「an integer is required」错误 #

你提到用生成器推导式时出现这个错误，大概率是把itertools.product的repeat参数写错了——这个参数必须是整数，用来指定生成的组合长度。比如如果你不小心把repeat=length写成了repeat=某个字符（比如repeat='A'），Python就会报错，因为它需要一个整数来确定组合长度。

关于拼接itertools.product的返回结果 #

itertools.product返回的是一个迭代器，每个元素是一个元组（比如长度2的组合会是('A', 'B')这样的形式）。要把这些元组转成字符串，直接用''.join()就可以。

如果想更高效地拼接不同长度的组合（尤其是字符集大、长度多的时候，不想一次性占用太多内存），可以用itertools.chain来拼接多个迭代器：

import itertools

chars = ['A', 'B', 'C']
target_lengths = [1, 2]

# 先为每个长度生成对应的迭代器
length_iterators = (itertools.product(chars, repeat=l) for l in target_lengths)
# 用chain把所有迭代器串起来，再逐个转成字符串
all_combinations = [''.join(item) for item in itertools.chain(*length_iterators)]

这种方式不会提前把所有组合都存到内存里，而是按需迭代，在处理大规模组合时更友好。

最后补充个小扩展 #

如果想生成从长度1到N的所有组合，只需要把target_lengths改成range(1, N+1)就行，比如生成长度1到3的组合：

target_lengths = range(1, 4)  # Python2.7里range返回列表，刚好能用

内容的提问来源于stack exchange，提问作者user3.1415927