Python生成文本行全排列出错求助：输出为字符而非单词

问题背景 #

给定文本文件内容如下：

1. El Paso there Yakima Sacramento Riverside Fresno Rialto Denver Fordham
2. Westminster Austin Tiverton Greensboro Riverside Fresno Rialto Denver Fordham
3. Westminster Lakeview Miami Tiverton Riverside Fresno Rialto Denver Fordham
4. Lakeview Austin Tiverton Greensboro Riverside Fresno Rialto Denver Fordham
5. Easton Yakima Miami St. Louis Riverside Fresno Rialto Denver Fordham

需要生成每行9个元素的全排列（n=9，k=9），预期得到1814400行结果（362880×5），但现有代码输出字符排列，不符合需求。

错误代码 #

import itertools

with open('12round.txt') as file:
    for line in file:
        perms = itertools.permutations(line, 9)
        for i in perms:
            print(' '.join(i))

错误原因 #

1. 字符级排列而非元素级：itertools.permutations()接收字符串时，会将其拆分为单个字符处理，而非按空格分割后的独立元素。
2. 未处理序号：每行开头的x. 部分未被剔除，导致序号字符也被纳入排列范围。

修正后的代码 #

以下代码会正确处理每行元素，生成预期的全排列，并将结果写入文件（避免大量输出卡顿）：

import itertools

# 打开输入文件和输出文件
with open('12round.txt', 'r') as infile, open('permutations_output.txt', 'w') as outfile:
    for line in infile:
        # 去掉首尾空白，分割序号和内容（只分割一次）
        stripped_line = line.strip()
        if not stripped_line:
            continue
        _, content = stripped_line.split('. ', 1)
        
        # 分割内容为元素列表
        elements = content.split()
        
        # 处理带空格的元素（比如第一行的"El Paso"），合并前两个元素得到9个元素
        if len(elements) > 9:
            elements = [' '.join(elements[:2])] + elements[2:]
        
        # 生成9个元素的全排列并写入文件
        for permutation in itertools.permutations(elements, 9):
            outfile.write(' '.join(permutation) + '\n')

说明 #

• 针对第一行中El Paso这类带空格的元素，代码加入了合并前两个元素的逻辑，确保最终元素数量为9个，匹配全排列需求。
• 将结果写入文件而非直接打印，避免终端因大量输出出现卡顿或崩溃。
• 9个元素的全排列数量为9! = 362880，5行总计362880×5 = 1814400行，完全符合预期。

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