如何在 Python Pandas 中從列名稱獲取列索引?


若要在 Python Pandas 中從列名稱獲取列索引,我們可以使用 **get_loc()** 方法。

步驟 −

  • 建立二維、長度可變、可能具有異構表格資料的 **df**。
  • 列印輸入 DataFrame **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"Columns in the given DataFrame: ", columns

column_name = "z"
column_index = columns.get_loc(column_name)
print"Index of the column ", column_name, " is: ", column_index

column_name = "x"
column_index = columns.get_loc(column_name)
print"Index of the column ", column_name, " is: ", column_index

column_name = "y"
column_index = columns.get_loc(column_name)
print"Index of the column ", 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

Columns in the given DataFrame: Index(['x', 'y', 'z'],
dtype='object')

Index of the column z is: 2
Index of the column x is: 0
Index of the column y is: 1

更新時間:2021-09-14

12K+ 瀏覽

開啟你的職業生涯

完成課程即可獲得認證

開始吧
廣告