如何用Python判断CSV非表头行并存入mailing_list变量

下面提供两种实用的实现方式，帮你完成需求：

方法一：通过行计数器判断表头 #

这种方式适合需要同时追踪行号，或者需要对表头做额外处理的场景：

import csv

def read_mailing_list_file():
    # 初始化空列表存储非表头数据行
    mailing_list = []
    with open('mailing_list.csv', 'r') as csv_file:
        file_reader = csv.reader(csv_file)
        line_count = 0
        for row in file_reader:
            if line_count == 0:
                # 第一行为表头，跳过并更新计数器
                line_count += 1
                continue
            # 将非表头行追加到列表
            mailing_list.append(row)
            line_count += 1
    return mailing_list

# 测试调用
mailing_data = read_mailing_list_file()
for row in mailing_data:
    print(row)

关键说明： #

• 修正了原代码中直接读取整个文件的错误操作，改用空列表初始化mailing_list，避免重复读取和资源泄漏
• 用line_count标记当前行号，第一行（line_count == 0）判定为表头，直接跳过
• with语句会自动管理文件资源，无需手动关闭文件

方法二：直接跳过表头行（更简洁） #

利用csv.reader的迭代器特性，直接取出表头行，后续遍历的都是数据行：

import csv

def read_mailing_list_file():
    mailing_list = []
    with open('mailing_list.csv', 'r') as csv_file:
        file_reader = csv.reader(csv_file)
        # 读取并跳过表头行（如果需要使用表头，可将header变量保存下来）
        header = next(file_reader)
        # 遍历剩余所有数据行并追加到列表
        for row in file_reader:
            mailing_list.append(row)
    return mailing_list

# 测试调用
mailing_data = read_mailing_list_file()
for row in mailing_data:
    print(row)

关键说明： #

• next(file_reader)会直接取出迭代器的第一个元素（表头行），之后的循环从第二行开始
• 代码更简洁，不需要维护行计数器，适合不需要处理表头的场景

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