Cómo dibujar un gráfico en Python donde hay dos variables x, una de las cuales necesita ser calculada a partir de datos proporcionados?
Tengo un archivo CSV que incluye los resultados de una encuesta en la que se les preguntó a los usuarios su edad (0 a 100) y su estado de ánimo (0=feliz 1=medio 2=triste). Mi intención es hacer un gráfico de barras en Python utilizando matplotlib o cualquier otra biblioteca de gráficos con el número de personas de cada edad en el eje y y luego una barra triple en el eje x para mostrar cuántas personas tristes, felices y medias hay para cada edad. El problema es que en el archivo CSV no hay una columna que contenga directamente datos sobre el conteo del número total de personas de cada edad, número de personas felices de cada edad y número de personas tristes de cada edad, etc. Cualquier consejo sobre cómo abordar este problema sería de gran ayuda. La tabla a continuación muestra algunas líneas del archivo CSV. Gracias
| Edad | Nivel de ánimo |
|---|---|
| 12 | 0 |
| 83 | 1 |
| 55 | 2 |
davy.ai
Para abordar este problema, se puede utilizar la biblioteca
pandasen Python para crear un nuevo dataframe a partir del archivo CSV. Este dataframe se puede utilizar para crear una nueva columna que contenga el recuento de personas de cada edad. También se pueden crear nuevas columnas para el recuento de personas de cada edad con un nivel de ánimo específico.Por ejemplo, se puede utilizar la función
groupbypara agrupar los datos por edad y luego utilizar la funciónaggpara contar el número de personas de cada edad. También se puede utilizar la funciónaggpara contar el número de personas de cada edad con un nivel de ánimo específico.Una vez que se tenga el dataframe con el recuento de personas de cada edad y el recuento de personas de cada edad con un nivel de ánimo específico, se puede utilizar la biblioteca
matplotlibpara crear un gráfico de barras con el número de personas de cada edad en el eje y y luego una barra triple en el eje x para mostrar cuántas personas tristes, felices y de ánimo medio hay para cada edad.