CSV文件处理程序保存后与原文件结构不一致问题求助

问题根源分析 #

你的代码存在几个关键错误，直接导致保存后的文件结构异常：

• 仅写入最后一行数据：循环结束后i指向最后一条记录，f.write(self.original_data[i])只把最后一行写入文件，前面修改的所有行都丢失了。
• 重复写入导致内容覆盖：先通过文件句柄写入修改后的original_data，紧接着调用original_df.to_csv，后者会完全覆盖前者的内容，等于前面的修改白做了。
• 手动拆分CSV字段风险：用split(',')拆分CSV行，如果原字段包含逗号（比如被引号包裹的内容），会错误拆分字段，直接破坏原结构。
• 文件格式选择混淆：保存对话框指定的是Text files (*.txt)，但实际要保存CSV，容易导致格式识别和兼容性问题。

修复方案（推荐：直接用DataFrame处理） #

放弃手动修改原始文本行的方式，利用pandas的CSV处理能力直接操作DataFrame，既能保证格式一致，又能避免手动解析的错误：

import csv  # 需要导入csv模块来设置引号规则

def save_all(self):
    if self.selected_column is None:
        QMessageBox.critical(self, "Error", "No column selected")
        return

    # 修正保存对话框为CSV格式
    file, _ = QFileDialog.getSaveFileName(self, "Save CSV file", "", "CSV files (*.csv)")
    if not file:
        return

    # 直接更新原始DataFrame的目标列
    self.original_df[self.selected_column] = self.df[self.selected_column]

    # 保存时匹配原文件的CSV格式规则
    self.original_df.to_csv(
        file,
        index=False,
        header=True,
        encoding='utf-8-sig',
        quoting=csv.QUOTE_ALL,  # 原文件字段带引号则用这个，按需调整为csv.QUOTE_MINIMAL等
        sep=',',  # 原文件的分隔符，默认逗号，若用其他符号需修改
        lineterminator='\n'  # 匹配原文件换行符，原文件是\r\n则改成这个
    )

替代方案（保留原始文本行处理） #

如果原文件有特殊格式无法用pandas直接处理，可以用csv模块安全解析和生成行，同时修复写入逻辑：

import csv

def save_all(self):
    if self.selected_column is None:
        QMessageBox.critical(self, "Error", "No column selected")
        return

    file, _ = QFileDialog.getSaveFileName(self, "Save CSV file", "", "CSV files (*.csv)")
    if not file:
        return

    column_index = self.df.columns.tolist().index(self.selected_column)
    # 循环更新每一行数据，用csv模块解析避免字段拆分错误
    for i, value in enumerate(self.df[self.selected_column].values, start=1):
        # 解析原始行
        reader = csv.reader([self.original_data[i]])
        line = next(reader)
        # 更新目标列
        line[column_index] = str(value)
        # 重新生成符合CSV规范的行
        writer = csv.writer([], quoting=csv.QUOTE_ALL)
        self.original_data[i] = ''.join(writer._row_to_str(line)).strip()

    # 写入所有行，而不是仅最后一行
    with open(file, 'w', encoding='utf-8-sig') as f:
        f.write('\n'.join(self.original_data))

关键注意事项 #

• 一定要匹配原文件的引号规则：如果原文件只有含特殊字符的字段带引号，把quoting改成csv.QUOTE_MINIMAL；如果所有字段都带引号用csv.QUOTE_ALL。
• 确认原文件的分隔符和换行符，如果不是默认的逗号和\n，要在to_csv或csv.writer里对应设置。

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