es.davy.ai

Preguntas y respuestas de programación confiables

¿Tienes una pregunta?

Si tienes alguna pregunta, puedes hacerla a continuación o ingresar lo que estás buscando.

Cómo almacenar después de agrupar

¿Cómo puedo utilizar la columna después de utilizar groupby? Digamos que

x = df_new.groupby('industry')['income'].mean().sort_values(ascending = False)

daría:

"industry"        
telecommunications        330
crypto.                   100
gas                       100
.
.
.

Me gustaría guardar el nombre de la industria con el ingreso más alto en una variable y utilizarlo para otras consultas, así que aquí sería “telecommunications”. Pero hacer x[0] daría 330…

También, por favor, recomiende una mejor redacción para esta pregunta… no sé los términos adecuados.

Tags:  , ,

Answer

  1. Avatar for davy.ai

    Después de usar groupby, puedes acceder a los valores únicos de la(s) columna(s) por las que agrupaste utilizando el atributo groups del objeto resultante. En el ejemplo dado, x.groups te daría un objeto similar a un diccionario donde las claves son los nombres únicos de las industrias y los valores son los índices correspondientes de estas industrias en el dataframe original. Para acceder al nombre de la industria con el ingreso más alto, puedes hacer lo siguiente:

    top_industry = x.index[0]
    

    Esto te dará "telecomunicaciones".

    En cuanto a una mejor redacción para la pregunta, podrías preguntar: “¿Cómo puedo acceder a las etiquetas de grupo (es decir, los valores únicos) después de usar groupby en pandas?”

Comments are closed.