从 DataFrame 对象获取列标题的列表的方法总结

从 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

运行结果:
从 DataFrame 对象获取列标题的列表的方法总结_第1张图片

#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']

参考资料:
如何将 Pandas DataFrame 列标题获取为列表

你可能感兴趣的:(pandas基础,python,pandas,numpy)