In this article we will see how to get column index from column name of a Dataframe. We will use Dataframe.columns attribute and Index.get_loc method of pandas module together.
Syntax: DataFrame.columns
Return: column names index
Syntax: Index.get_loc(key, method=None, tolerance=None)
Return: loc : int if unique index, slice if monotonic index, else mask
Code: Let’s create a Dataframe:
Python3
# import pandas libraryimport pandas as pd # dictionaryrecord = {'Math': [10, 20, 30, 40, 70], 'Science': [40, 50, 60, 90, 50], 'English': [70, 80, 66, 75, 88]} # create a dataframedf = pd.DataFrame(record) # show the dataframeprint(df) |
Output:
Dataframe
Example 1: Get a index number of “Science” column.
Python3
# import pandas libraryimport pandas as pd # dictionaryrecord = {'Math': [10, 20, 30, 40, 70], 'Science': [40, 50, 60, 90, 50], 'English': [70, 80, 66, 75, 88]} # give column namecol_name = "Science" # find the index noindex_no = df.columns.get_loc(col_name) print("Index of {} column in given dataframe is : {}".format(col_name, index_no)) |
Output :
index number of Science column
Example 2: Get a index number of “English” column.
Python3
# import pandas libraryimport pandas as pd # dictionaryrecord = {'Math': [10, 20, 30, 40, 70], 'Science': [40, 50, 60, 90, 50], 'English': [70, 80, 66, 75, 88]} # create a dataframedf = pd.DataFrame(record) # give column namecol_name = "English" # find the index noindex_no = df.columns.get_loc(col_name) print("Index of {} column in given dataframe is : {}".format(col_name, index_no)) |
Output :
index number of English column
