从一个给定的Pandas数据框架的列名中获取列索引

在这篇文章中，我们将看到如何从Dataframe的列名中获得列索引。我们将同时使用Dataframe.columns属性和pandas模块的Index.get_loc方法。

语法: DataFrame.columns

返回：列名索引

语法: Index.get_loc(key, method=None, tolerance=None)

返回： loc : 如果是唯一的索引，则为int；如果是单调的索引，则为slice；否则为mask

代码：让我们创建一个数据框架。

# import pandas library
import pandas as pd
  
# dictionary
record = {'Math': [10, 20, 30, 
                   40, 70],
          'Science': [40, 50, 60,
                      90, 50], 
          'English': [70, 80, 66, 
                      75, 88]}
  
# create a dataframe
df = pd.DataFrame(record)
  
# show the dataframe
print(df)

输出:

从一个给定的Pandas数据框架的列名中获取列索引

例子1：获得 “科学 “列的索引号。

# import pandas library
import pandas as pd
  
# dictionary
record = {'Math': [10, 20, 30, 40, 70],
          'Science': [40, 50, 60, 90, 50], 
          'English': [70, 80, 66, 75, 88]}
  
# give column name
col_name = "Science"
  
# find the index no
index_no = df.columns.get_loc(col_name)
  
print("Index of {} column in given dataframe is : {}".format(col_name, index_no))

输出 :

从一个给定的Pandas数据框架的列名中获取列索引

例子2：获得 “英语 “列的索引号。

# import pandas library
import pandas as pd
  
# dictionary
record = {'Math': [10, 20, 30,
                   40, 70],
          'Science': [40, 50, 60,
                      90, 50], 
          'English': [70, 80, 66,
                      75, 88]}
  
# create a dataframe
df = pd.DataFrame(record)
  
# give column name
col_name = "English"
  
# find the index no
index_no = df.columns.get_loc(col_name)
  
print("Index of {} column in given dataframe is : {}".format(col_name, index_no))