无需存储全量数据，如何高效更新CSV文件中的单行？

原代码需要把整个CSV文件加载到内存，修改后再全量写入，面对大文件时内存占用高、效率低。可以通过逐行读写+临时文件的方式解决，不用存储整个文件的内容：

核心思路 #

• 打开原CSV文件和一个临时文件
• 逐行读取原文件内容，找到目标用户对应的行时修改登录次数
• 将处理后的每行（不管修改与否）写入临时文件
• 用临时文件替换原文件（原子操作，避免文件损坏）

实现代码 #

import csv
import os

def increment_login_count(username: str) -> int:
    """
    更新用户的登录次数（无需加载整个CSV到内存）并返回新值
    
    参数:
        username (str): 需要更新登录次数的用户
    
    返回值:
        login_count (int): 更新后的登录次数，未找到用户则返回0
    """
    login_count = 0
    temp_file = "users_temp.csv"
    
    # 逐行读取原文件，写入临时文件
    with open("users.csv", "r", encoding="utf-8", newline="") as infile, \
         open(temp_file, "w", encoding="utf-8", newline="") as outfile:
        
        reader = csv.reader(infile)
        writer = csv.writer(outfile)
        
        for row in reader:
            # 找到目标用户行
            if row[1] == username:
                login_count = int(row[4]) + 1
                row[4] = str(login_count)  # 转成字符串保持CSV格式统一
            writer.writerow(row)
    
    # 用临时文件替换原文件（原子操作，更安全）
    os.replace(temp_file, "users.csv")
    
    return login_count

注意事项 #

• 使用os.replace()而非os.rename()，前者在跨文件系统场景下也能保证原子性，避免原文件损坏
• 确保CSV中登录次数列是可转换为整数的字符串格式
• 若需处理文件被占用等极端情况，可添加对应的异常捕获逻辑

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