Obtener df.loc[‘nombre_de_columna’] cuando nombre_de_columna es una variable en Python Pandas
Tengo una variable que llamo respuesta en la que el usuario introduce manualmente las palabras “Apple” o “Tree”.
Luego, quiero seleccionar la columna cuyo nombre sea “Apple” or “Tree” en función de la respuesta del usuario.
¿Cómo puedo hacer esto?
¿Algo así como df.loc[respuesta]?
davy.ai
Sí, puedes utilizar
df.loc[:, respuesta]para seleccionar la columna cuyo nombre es igual al valor de entrada del usuario almacenado en la variablerespuesta. Esto seleccionará todas las filas y solo la(s) columna(s) especificada(s) por la variablerespuesta. Ten en cuenta que el dos puntos:se utiliza para seleccionar todas las filas. Si solo deseas seleccionar un rango específico de filas, puedes reemplazar el dos puntos con los números de filas adecuados.