博客

python获取Pandas列名的几种方法

  • Post category:Python

想要获取Pandas DataFrame列名，可以使用以下几种方法：

1. 打印列名

DataFrame的列名可以通过DataFrame的columns属性来获取:

import pandas as pd 

df = pd.DataFrame({'A': [1, 2, 3], 
                   'B': [4, 5, 6], 
                   'C': [7, 8, 9]}) 

print(df.columns)

输出结果为:

Index(['A', 'B', 'C'], dtype='object')

2. 使用keys()方法

keys()方法可以直接返回DataFrame对象的列名：

import pandas as pd 

df = pd.DataFrame({'A': [1, 2, 3], 
                   'B': [4, 5, 6], 
                   'C': [7, 8, 9]}) 

print(df.keys())

输出结果为:

Index(['A', 'B', 'C'], dtype='object')

3. 转换为列表

将DataFrame的columns属性或keys()方法获取到的index对象转换为列表，也可以得到DataFrame对象的列名:

import pandas as pd 

df = pd.DataFrame({'A': [1, 2, 3], 
                   'B': [4, 5, 6], 
                   'C': [7, 8, 9]}) 

#使用df.columns属性
print(list(df.columns)) 

#使用df.keys方法
print(list(df.keys()))

输出结果为:

['A', 'B', 'C']
['A', 'B', 'C']

以上就是获取Pandas DataFrame列名的几种方法，可以根据实际情况进行选择。

Tags: