如何从熊猫的索引输出中提取列表?

时间:2017-12-11 10:02:25

标签: python list pandas csv indexing

我想阅读csv文件的列(a,b和c)。我使用以下代码: -

tt=pd.read_csv('abc.csv', nrows=1,sep='|').columns

输出是一个索引,如下所示。

Index(['a', 'b', 'c'], dtype='object')

如何从上面的索引中提取列表['a','b','c']。有没有其他更优雅的方法将csv文件的列标题读取到列表而不加载整个csv文件?

1 个答案:

答案 0 :(得分:0)

如评论中所述。 tolist函数可用于将索引转换为列表

> import pandas as pd 
> df = pd.DataFrame({'a':[1,2,3],'b':['a','b','c']})    
> df.columns.tolist()