如何用Python的csv.DictReader读取可能含首行表头的CSV文件？

解决思路 #

先检查文件第一行是不是预设的固定表头col1,col2，再针对性地用csv.DictReader读取，既避免把表头当成数据，也解决无表头时返回空列表的问题。

代码实现 #

import csv

def read_target_csv(file_path):
    with open(file_path, 'r', newline='') as f:
        # 用csv.reader读取第一行，避免字段含逗号时判断出错
        csv_reader = csv.reader(f)
        first_row = next(csv_reader)
        # 重置文件指针到开头，让DictReader能从头读取
        f.seek(0)
        
        expected_headers = ['col1', 'col2']
        if first_row == expected_headers:
            # 存在预设表头，直接用DictReader自动识别表头
            reader = csv.DictReader(f)
        else:
            # 无表头，指定固定字段名
            reader = csv.DictReader(f, fieldnames=expected_headers)
        
        return list(reader)

测试效果 #

• 带表头的CSV文件：

  col1,col2
  foo,bar
  

  返回结果：[{'col1': 'foo', 'col2': 'bar'}]，符合预期。

• 不带表头的CSV文件：

  foo,bar
  

  返回结果：[{'col1': 'foo', 'col2': 'bar'}]，满足需求。

补充说明 #

如果CSV字段包含逗号（比如"hello,world",bar），用普通字符串分割判断表头会出错，代码里用csv.reader读取第一行的方式，能正确解析带引号的特殊字段，兼容性更强。

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