如何仅从 csv 文件(熊猫)加载列名?

Posted

技术标签:

【中文标题】如何仅从 csv 文件(熊猫)加载列名?【英文标题】:How to load Only column names from csv file (Pandas)? 【发布时间】:2017-10-21 22:16:24 【问题描述】:

我有一个大的 csv 文件,不想将它完全加载到我的内存中,我只需要从此 csv 文件中获取列名。怎么加载清楚?

【问题讨论】:

【参考方案1】:

试试这个:

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

【讨论】:

【参考方案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')

【讨论】:

以上是关于如何仅从 csv 文件(熊猫)加载列名?的主要内容,如果未能解决你的问题,请参考以下文章

如何将熊猫数据添加到现有的 csv 文件中?

使用熊猫重复加载 csv [重复]

如何将csv数据加载到matplotlib?

如何在循环中附加多个熊猫数据框?

加载文件项后如何仅从 dataTransfer 调度 FileList

如何将csv文件转换为numpy数组格式?