我想知道是否有办法只读取csv文件中的标题信息而不加载整个文件。我正在处理大型csv文件,而不是不必加载整个文件。
谢谢!
答案 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
)。