从一个给定的Pandas数据框架的列名中获取列索引

  • Post category:Python

获取Pandas数据框架的列索引方法如下:

  1. 使用df.columns方法获取所有列名列表。

  2. 使用df.columns.get_loc()方法得到指定列名的索引。

以下是操作演示和示例代码:

  1. 创建一个示例数据框架df:
import pandas as pd
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
  1. 查看数据框架的所有列名:
print(df.columns)

输出:

Index(['A', 'B'], dtype='object')
  1. 得到指定列名的索引,例如获取列名为’B’的索引:
index = df.columns.get_loc('B')
print(index)

输出:

1

这里的输出结果1表示’B’列在数据框架的第二列,对应的列索引为1。

通过以上步骤,我们得到了从Pandas数据框架中获取某一列名对应的列索引的实现方法,可以在需要时灵活应用。