我想阅读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文件?
答案 0 :(得分:0)
如评论中所述。 tolist函数可用于将索引转换为列表
> import pandas as pd
> df = pd.DataFrame({'a':[1,2,3],'b':['a','b','c']})
> df.columns.tolist()