如何只从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')