How to get column index from column name in Python Pandas?

Python Server Side Programming Programming

To get column index from column name in Python Pandas, we can use the get_loc() method.

Steps −

Create a two-dimensional, size-mutable, potentially heterogeneous tabular data, df.
Print the input DataFrame, df.
Find the columns of DataFrame, using df.columns.
Print the columns from Step 3.
Initialize a variable column_name.
Get the location, i.e., of index for column_name.
Print the index of the column_name.

Example −

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

Output

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

Rishikesh Kumar Rishi

Updated on: 2021-09-14T11:45:05+05:30

12K+ Views

Kickstart Your Career

Get certified by completing the course

Get Started