有沒有辦法只讀python中的csv文件頭？

Question

我想知道是否有一種方法只從csv文件中讀取頭信息而不加載整個文件。我正在處理大型csv文件，而不必加載整個事情。

謝謝！

Answer 1

反正你通常不會讀整個文件。只需創建一個CSV閱讀器並在其上撥打next()即可。

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

Answer 2

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），它將僅輸出數據。