Obtenga el índice de la columna del nombre de la columna de un Pandas DataFrame dado

En este artículo, veremos cómo obtener el índice de columna del nombre de columna de un marco de datos. Usaremos el atributo Dataframe.columns y el método Index.get_loc del módulo pandas juntos.

Sintaxis: DataFrame.columns

Retorno: índice de nombres de columna

Sintaxis: Index.get_loc(clave, método=Ninguno, tolerancia=Ninguno)

Devuelve: loc: int si es un índice único, rebanada si es un índice monótono, si no es una máscara

Código: Vamos a crear un marco de datos:

# 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)

Producción:

Marco de datos

Ejemplo 1:  obtenga un número de índice de la columna «Ciencia».

# 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))

Producción :

número de índice de la columna de ciencia

Ejemplo 2: obtenga un número de índice de la columna «Inglés».

# 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))

Producción :

número de índice de la columna en inglés