如何从 Python Pandas 中的列名获取列索引?

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

步骤 -

  • 创建二维、大小可变、潜在异构的表格数据df。

  • 打印输入数据帧df。

  • 使用df.columns查找 DataFrame 的列。

  • 打印步骤 3 中的列。

  • 初始化变量column_name。

  • 获取位置,即column_name的索引。

  • 打印column_name的索引。

示例 -

import pandas as pd

df = pd.DataFrame(

   {

      "x": [5, 2, 7, 0],

      "y": [4, 7, 5, 1],

      "z": [9, 3, 5, 1]

   }

)

print"Input DataFrame 1 is:\n", df

columns = df.columns

print"给定 DataFrame 中的列: ", columns

column_name = "z"

column_index = columns.get_loc(column_name)

print"列的索引 ", column_name, " is: ", column_index

column_name = "x"

column_index = columns.get_loc(column_name)

print"列的索引 ", column_name, " is: ", column_index

column_name = "y"

column_index = columns.get_loc(column_name)

print"列的索引 ", column_name, " is: ", column_index

输出结果
Input DataFrame 1 is:

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 is: 2

列的索引 x is: 0

列的索引 y is: 1

以上是 如何从 Python Pandas 中的列名获取列索引? 的全部内容, 来源链接: utcz.com/z/362053.html

回到顶部