如何在 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
廣告