文件去空行方法末尾新增空行问题修复及代码简化求助

为什么末尾会出现额外空行？ #

通常这个问题出在写入文件时的换行符处理上。举个例子，假设你原来的代码大概是这两种常见写法之一：

第一种是直接遍历写入过滤后的行：

with open("input.txt", "r") as infile:
    non_empty_lines = [line for line in infile if line.strip()]

with open("output.txt", "w") as outfile:
    for line in non_empty_lines:
        outfile.write(line)

第二种是用print()函数写入：

with open("output.txt", "w") as outfile:
    for line in non_empty_lines:
        print(line, file=outfile)

问题根源：

1. 第一种：如果原文件的最后一行本身带有换行符\n，写入后这个换行符会留在文件末尾，形成空行；
2. 第二种：print()函数默认会在输出内容后追加一个换行符，而原行可能已经包含换行符，相当于重复添加，最后就多了一行空行。

如何修复？ #

针对不同需求，有两种简单的修复方式：

方式1：统一处理换行符（适合需要标准化格式的场景） #

先把每行的换行符去掉，再用\n连接所有非空行，最后一次性写入——这样只有行之间会有换行，末尾不会多出空行：

with open("input.txt", "r") as infile:
    # 过滤空行并移除每行末尾的换行符
    non_empty = [line.strip() for line in infile if line.strip()]

with open("output.txt", "w") as outfile:
    outfile.write("\n".join(non_empty))

方式2：保留原行格式（适合需要保留原文本缩进/末尾空格的场景） #

如果想保留原行的原始格式，只去掉最后的多余换行，可以这样写：

with open("input.txt", "r") as infile:
    non_empty_lines = [line for line in infile if line.strip()]

with open("output.txt", "w") as outfile:
    if non_empty_lines:
        # 先写入除最后一行外的所有内容
        outfile.writelines(non_empty_lines[:-1])
        # 最后一行单独写入，去掉末尾可能的换行符
        outfile.write(non_empty_lines[-1].rstrip("\n"))

代码能不能简化？ #

当然可以！可以把读写操作合并成一段更简洁的代码，用生成器表达式减少内存占用（尤其适合大文件）：

简化版1（标准化格式） #

with open("input.txt", "r") as infile, open("output.txt", "w") as outfile:
    # 用生成器过滤空行，join自动处理换行，无末尾空行
    outfile.write("\n".join(line.strip() for line in infile if line.strip()))

简化版2（保留原格式） #

with open("input.txt", "r") as infile, open("output.txt", "w") as outfile:
    # 生成器过滤空行，最后去掉末尾多余换行
    outfile.write("".join(line for line in infile if line.strip()).rstrip("\n"))

这样既简洁高效，又完美解决了末尾空行的问题。

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