从 DataFrame 对象获取列标题的列表的方法总结:
import pandas as pd
import numpy as np
dates=['April-10', 'April-11', 'April-12', 'April-13','April-14','April-16']
sales=[200,300,400,200,300,300]
prices=[3, 1, 2, 4,3,2]
df = pd.DataFrame({'Date':dates ,
'Sales':sales ,
'Price': prices})
df
运行结果:
#1.迭代 DataFrame 对象本身,返回的是一个数据框
column_names = pd.DataFrame(df)
#打印列名
for column_name in column_names:
print(column_name)
#将数据框转化为list并打印
print(list(column_names ))
#运行结果:
Date
Sales
Price
['Date', 'Sales', 'Price']
#2. 使用DataFrame.columns 属性访问其列再转换为list
df.columns.tolist()
#运行结果:
['Date', 'Sales', 'Price']
# 3.使用DataFrame.columns.values
print(list(df.columns.values))#性能次之
print(df.columns.values.tolist())#性能最好
#运行结果:
['Date', 'Sales', 'Price']
['Date', 'Sales', 'Price']