python获取Pandas列名的几种方法

本文汇总了几种获得CSV Excel 文件字段列表的的方法，一起来看看吧

下面几种方法的代码基于这样的测试数据：

import pandas as pd

file = r'E:\数据分析\pandas_02\data.csv'
df = pd.read_csv(file,engine='python')
df

 

1.通过columns字段获取，返回一个numpy型的array

fields = df.columns.values
print(fields)

['品牌' '价格' '好评数']

2.直接使用 list 关键字，返回一个list

fields = list(df)
print(fields)

['品牌', '价格', '好评数']

3.通过 tolist(), 或者 list（array） 转换为list

fields = df.columns.tolist()
print(fields)

['品牌', '价格', '好评数']

4.链表推倒式

fields = [column for column in df]
print(fields)

['品牌', '价格', '好评数']

总之，条条大路通罗马，总有一款适合您。