python - 有没有办法只从 python 中的 csv 文件中读取标题？

Question

我想知道是否有一种方法可以只从 csv 文件中读取标题信息而不加载整个文件。我正在处理大型 csv 文件，并且不想加载整个文件。

谢谢！

score 2 · Accepted Answer

with open(filename) as in_file:
    csv_reader = csv.reader(in_file)
    header = next(csv_reader)

这是有效的，因为csv.reader()返回一个生成器，而不是一个列表。它只会在next()被调用时输出数据（即通过使用next(csv_reader)或在 for 循环中使用它，如 for row in csv_reader）。

score 1 · Accepted Answer

无论如何，您通常不会阅读整个文件。只需创建 CSV 阅读器并调用next()一次。

import csv
f = open('myfile.csv')
reader = csv.reader(f)
header = reader.next()

2 回答 2