如何生成含重复数字的数字组合？Python itertools使用疑问

你当前使用的permutations函数的特性就是生成无重复元素的排列，它会从输入列表中选取不重复的元素进行排列，所以自然无法生成包含重复数字的组合（比如245325、111222这类）。

要生成允许重复数字的所有6位数字组合，应该使用itertools.product——它会生成输入可迭代对象的笛卡尔积，允许同一元素被多次选取。

修改后的代码 #

from itertools import product

# 生成所有6位数字组合（允许重复数字），每个位置可选0-9
comb = product([0,1,2,3,4,5,6,7,8,9], repeat=6)

# 使用with语句自动管理文件，更安全
with open("code.txt", "w") as f:
    for nums_tuple in comb:
        # 直接将元组转成字符串，比原来的translate更简洁
        num_str = ''.join(map(str, nums_tuple))
        print(num_str)
        f.write(num_str + "\n")

关键说明 #

• product(iterable, repeat=n)：会生成长度为n的所有可能组合，每个位置的元素都可以从输入的可迭代对象中重复选取，完全符合你需要的重复数字场景。
• 字符串处理优化：用''.join(map(str, nums_tuple))直接把数字元组转成连续字符串，替代原来复杂的translate操作，代码更易读且高效。
• 文件操作优化：使用with语句打开文件，无需手动调用close()，避免因程序异常导致文件未正确关闭的问题。

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