如何从csv文件(Pandas)加载只有列名？

Question

我有一个大的csv文件,不想将它完全加载到我的内存中,我需要从这个csv文件中只获取列名.如何清楚地加载它？

Answer 1

试试这个:

pd.read_csv(file_name, nrows=1).columns.tolist()

Answer 2

如果你传递nrows=0给read_csv那么它只会加载列行:

In[8]:
import pandas as pd
import io
t="""a,b,c,d
0,1,2,3"""
pd.read_csv(io.StringIO(t), nrows=0)

Out[8]: 
Empty DataFrame
Columns: [a, b, c, d]
Index: []

之后,访问属性.columns将为您提供以下列:

In[10]:
pd.read_csv(io.StringIO(t), nrows=0).columns

Out[10]: Index(['a', 'b', 'c', 'd'], dtype='object')