Python: Pandas Dataframe seleccionar fila por índice dado
Tengo un dataframe de Pandas de este tipo:
data = {'Index':['1a','2a','3a','4a'], 'col1':[20.1,20.2,20.3,20.4], 'col2': [30.2,30.5,30.7,30.5]}
df1 = pd.DataFrame(data)
df1 = df1.set_index('Index')
print(df1)
Salida:
col1 col2
Index
1a 20.1 30.2
2a 20.2 30.5
3a 20.3 30.7
4a 20.4 30.5
Ahora mi objetivo es seleccionar la fila con un índice específico, por ejemplo 2a, por lo que la salida debería ser:
col1 col2
Index
2a 20.2 30.5
Lo intenté con el siguiente comando (ver por ejemplo este tutorial: link):
df1.loc[df1['Index'] == 2]
Pero por alguna razón no funciona, ¿por qué?
davy.ai
La razón por la cual el comando anterior no funciona es porque está buscando una fila donde el valor en la columna “Índice” sea igual a 2, mientras que los valores de índice reales son “2a”, “3a”, y así sucesivamente.
Para seleccionar la fila con el índice “2a”, puede utilizar el siguiente comando:
Esto devolverá la salida deseada: