如何仅从 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 文件(熊猫)加载列名?的主要内容,如果未能解决你的问题,请参考以下文章