Python替换字符串字符为所有组合后追加列表写入文件问题求助

问题原因 #

现有代码的核心错误如下：

• 递归调用replaceX的逻辑写在生成器的循环内部，会触发无限递归，无法正常生成替换结果
• replaceX是生成器函数，直接使用*replaceX(...)作为append的参数会抛出参数数量异常
• 始终没有在函数外部主动触发replaceX生成结果，keyList默认是空列表，自然写入空内容

修正后可运行代码 #

import itertools

def replaceX(s, chars):
    # 计算X的数量，生成对应长度的所有字符组合
    for combo in itertools.product(chars, repeat=s.count("X")):
        combo_iter = iter(combo)
        # 逐个替换X为组合里的对应字符
        yield ''.join(char if char != 'X' else next(combo_iter) for char in s)

if __name__ == "__main__":
    origin_str = "ABCXXABXXABCDEF\n"
    replace_chars = "ABCDEF"
    # 遍历生成器把所有结果存入列表
    keyList = list(replaceX(origin_str, replace_chars))

    # 写入文件
    with open("key-file.txt", "w", encoding="utf-8") as outFile:
        outFile.writelines(keyList)
    
    # 可打印数量校验：4个X总共有6^4=1296条结果
    print(f"共生成{len(keyList)}条结果，已写入key-file.txt")

改动说明 #

• 移除了函数内部错误的递归调用和列表追加逻辑，生成器仅负责生成替换后的字符串，职责更清晰
• 统一在主逻辑中遍历生成器，将所有结果转成列表，避免无限递归和参数异常
• 改用with语法操作文件，自动处理文件关闭，避免资源泄漏
• 补充了字符编码声明，避免不同系统下写入乱码

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