Python批量提取CSV文件第二行数据失败问题求助

你的核心问题是当前代码读取了整个文件内容，没有针对性提取第二行。以下是几种直接可行的修改方案，你可以根据需求选择：

方案一：使用itertools.islice（你已导入该模块） #

利用islice直接定位到第二行，无需读取全部文件，效率较高：

import os
from tkinter import filedialog
from itertools import islice

# 选择文件夹
root = filedialog.Tk()
root.withdraw()
target_folder = filedialog.askdirectory()
target_file = filedialog.asksaveasfilename(defaultextension=".txt")  # 改为保存文件，符合写入需求

content = []

for subdir, dirs, files in os.walk(target_folder):
    for filename in files:
        # 增加.csv后缀判断，避免误匹配非CSV文件
        if filename.lower().startswith('autopilot') and filename.endswith('.csv'):
            filepath = os.path.join(subdir, filename)
            with open(filepath, 'r', encoding='utf-8') as f:
                # islice(f, 1, 2) 表示从第2行（索引1）开始，取1行
                try:
                    second_line = next(islice(f, 1, 2)).strip()
                    print(f"{filename}: {second_line}")
                    content.append(second_line)
                except StopIteration:
                    print(f"{filename} 没有第二行，跳过")

# 将提取的内容写入目标文件
with open(target_file, 'w', encoding='utf-8') as out_f:
    out_f.write('\n'.join(content))

方案二：直接读取行列表 #

适合文件较小的场景，简单直观：

# 其他部分同上，循环内代码修改为：
with open(filepath, 'r', encoding='utf-8') as f:
    lines = f.readlines()
    if len(lines) >= 2:
        second_line = lines[1].strip()
        print(f"{filename}: {second_line}")
        content.append(second_line)
    else:
        print(f"{filename} 不足两行，跳过")

方案三：使用csv模块处理 #

如果CSV文件包含复杂格式（比如带逗号的字段），用csv模块解析更可靠：

import csv

# 其他部分同上，循环内代码修改为：
with open(filepath, 'r', encoding='utf-8') as f:
    reader = csv.reader(f)
    next(reader)  # 跳过第一行
    try:
        second_row = next(reader)
        second_line = ','.join(second_row)  # 还原为CSV格式字符串
        print(f"{filename}: {second_line}")
        content.append(second_line)
    except StopIteration:
        print(f"{filename} 没有第二行，跳过")

关键修改说明 #

1. 替换f.read()为针对性读取第二行的逻辑，避免输出全部内容；
2. 增加文件行数不足的异常处理，防止程序报错；
3. 将target_file的askopenfilename改为asksaveasfilename，符合写入文件的实际需求；
4. 补充.csv后缀判断，避免匹配到非目标文件。

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