如何避免MemoryError并将itertools.combinations_with_replacement结果写入文件

嘿，这个问题太常见了——你遇到的MemoryError完全是因为一次性把所有组合加载到内存里了。combinations_with_replacement本身是个迭代器，它不会一次性生成所有结果，而是按需逐个产出。我们正好可以利用这个特性，把每个组合直接写入文件，而不是先转成list存到内存里。

具体实现代码 #

from itertools import combinations_with_replacement

# 生成你的目标数组
arr = list(range(495))

# 打开文件（'w'表示覆盖原有内容，若要追加用'a'）
with open('combinations_output.txt', 'w', encoding='utf-8') as output_file:
    # 逐个迭代组合，避免一次性加载全部
    for combo in combinations_with_replacement(arr, 10):
        # 将组合转为逗号分隔的字符串，每行写一个组合
        output_file.write(','.join(map(str, combo)) + '\n')

为什么这能解决问题？ #

• 迭代器combinations_with_replacement每次只生成一个组合，处理完就写入文件，内存里永远只保留当前这一个组合，不会积累海量结果。
• with语句会自动帮你管理文件的打开和关闭，避免资源泄漏。

重要提醒 #

你要做好心理准备：495个元素取10个的可重复组合数是天文数字（计算下来约为2.6×10^22），写入文件会花费极长的时间，而且最终文件大小会远超普通磁盘的容量。如果只是验证逻辑，建议先缩小数组规模（比如用10个元素取3个组合）测试。

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