有没有办法只在python中读取csv文件中的标题?

时间:2013-11-14 17:25:24

标签: python csv header

我想知道是否有办法只读取csv文件中的标题信息而不加载整个文件。我正在处理大型csv文件,而不是不必加载整个文件。

谢谢!

2 个答案:

答案 0 :(得分:1)

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

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

答案 1 :(得分:1)

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

这是有效的,因为csv.reader()会返回generator,而不是列表。它只会在调用next()时输出数据(即使用next(csv_reader)或在for循环中使用它,例如for row in csv_reader)。