获得 Pandas 数据框架的列名是非常简单的,只需要调用数据框架的 columns
属性即可。下面是一个具体的例子:
import pandas as pd
# 创建数据框架
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
# 获取列名
cols = df.columns
# 打印列名
print(cols)
输出结果为:
Index(['A', 'B'], dtype='object')
可以看到,columns
属性返回了一个 Index
对象,其中包含了所有的列名。如果只需要获取列名的列表,可以通过 tolist()
方法进行转换:
# 获取列名列表
col_list = df.columns.tolist()
# 打印列名列表
print(col_list)
输出结果为:
['A', 'B']
另外,如果需要获取某一列的列名,可以使用以下方式(假设要获取名为 'A' 的列名):
# 获取 'A' 列的列名
col_name = df.columns[0]
# 打印列名
print(col_name)
输出结果为:
A
因此,以上就是在 Pandas 数据框架中获得列名的完整攻略。
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:如何在Pandas数据框架中获得列名 - Python技术站