¿Cómo regresar 3 columnas encima y debajo de la celda especificada en un dataframe de pandas?

Answer

1. 

   davy.ai

   May 27, 2023 at 6:47 pm

   Para obtener la salida deseada, primero necesitamos filtrar el dataframe según el estado de entrada, que es Iowa en este caso. Luego, podemos rebanar el dataframe filtrado para obtener los 3 estados arriba y abajo de Iowa mediante los valores del índice. Aquí está el código para hacer eso:

   # Suponiendo que el dataframe está almacenado en una variable llamada "df"
   
   input_state = "Iowa"
   
   # Filtrar el dataframe según el estado de entrada
   filtered_df = df[df["States"] == input_state]
   
   # Obtener el índice del estado de entrada
   idx = filtered_df.index[0]
   
   # Rebanar el dataframe para obtener 3 estados arriba y abajo del estado de entrada
   output_df = df.iloc[max(idx-3, 0) : min(idx+4, len(df))]
   

   En el código anterior, primero filtramos el dataframe df según el estado de entrada usando la indexación booleana. Esto crea un nuevo dataframe filtered_df que solo contiene filas donde la columna States coincide con el estado de entrada.

   A continuación, obtenemos el valor de índice del estado de entrada del dataframe filtrado usando el atributo index y tomando el primer valor ya que solo tenemos una fila en el dataframe filtrado.

   Finalmente, rebanamos el dataframe original df utilizando los valores de índice calculados en función del estado de entrada. Usamos las funciones max y min para asegurarnos de que no salimos de los límites al obtener los 3 estados arriba y abajo del estado de entrada. El dataframe resultante output_df contiene la salida deseada: el nombre del país (EE. UU.) y 3 estados arriba y abajo de Iowa.

Comments are closed.