如何在Python Pandas中通过列名获取列索引？

要从Python Pandas中的列名获取列索引，可以使用 get_loc() 方法。

步骤 –

• 创建一个二维、可变大小、潜在异构表格数据， df 。
• 输出输入的DataFrame， df 。
• 使用 df.columns 找到DataFrame的列。
• 打印步骤3的列。
• 初始化一个变量 column_name 。
• 获取 column_name 的位置，即索引。
• 输出 column_name 的索引。

更多Pandas文章，请阅读：Pandas教程

示例 –

import pandas as pd

df = pd.DataFrame(
   {
      "x": [5, 2, 7, 0],
      "y": [4, 7, 5, 1],
      "z": [9, 3, 5, 1]
   }
)

print"输入的DataFrame 1为:\n", df
columns = df.columns
print"给定DataFrame中的列: ", columns

column_name = "z"
column_index = columns.get_loc(column_name)
print"列 ", column_name, " 的索引为: ", column_index

column_name = "x"
column_index = columns.get_loc(column_name)
print"列 ", column_name, " 的索引为: ", column_index

column_name = "y"
column_index = columns.get_loc(column_name)
print"列 ", column_name, " 的索引为: ", column_index

输出

输入的DataFrame 1为:
x y z
0 5 4 9
1 2 7 3
2 7 5 5
3 0 1 1

给定DataFrame中的列: Index(['x', 'y', 'z'],
dtype='object')

列 z 的索引为: 2
列 x 的索引为: 0
列 y 的索引为: 1