Agrupar datos
Parte de la sección Datos y gráficos del Journey de Machine Learning de Coddy. Lección 43 de 60.
¿Qué ciudad vende más? Un promedio de todas las filas no puede responderlo. Divide las filas en grupos, uno por ciudad, y después calcula el promedio de cada grupo:
Una línea hace ambas cosas:
shops = pd.DataFrame({
"city": ["Lyon", "Paris", "Lyon", "Nice", "Paris", "Lyon"],
"sales": [20, 35, 26, 18, 41, 23],
})
means = shops.groupby("city")["sales"].mean()
print(means)Esto imprime:
city
Lyon 23.0
Nice 18.0
Paris 38.0
Name: sales, dtype: float64groupby("city") crea los grupos, ["sales"] selecciona la columna, .mean() calcula el promedio de cada grupo. .sum(), .min() y .max() también funcionan.
Para clasificar las pilas, ordénalas: means.sort_values(ascending=False).index[0] es Paris, el más vendido.
Desafío
IntermedioUn club de corredores registra cada carrera en runs.csv: runner, distance_km, minutes y climb_m. El código inicial lee una de las columnas numéricas en column.
- Imprime la media de esa columna para cada corredor.
- Después, indica el corredor con la media más alta, como en la última línea de abajo.
Para un club con dos corredores y distance_km:
runner
Ana 6.1
Bo 8.4
Name: distance_km, dtype: float64
Top runner: Bo with 8.4Pruébalo tú mismo
import pandas as pd
runs = pd.read_csv("runs.csv")
column = input()
# Escribe tu código a continuación
Esta lección incluye un breve cuestionario. Empieza la lección para responderlo y registrar tu progreso.
Todas las lecciones de Datos y gráficos
4Dar forma a los datos
Cambiar la forma de los arraysUnir arraysNúmeros aleatoriosRepaso: simulación de dados2Arrays de NumPy
Creación de arraysForma y tamañoRangos de númerosIndexación y segmentaciónRepaso: cuadrícula de temperaturas5Gráficos de líneas y de dispersión
Etiquetas y títulosGráficos de dispersiónVarias líneas y leyendasColores y marcadoresFiguras y ejesRepaso: pasos semanales3Matemáticas con arrays
Operaciones matemáticas con arrays completosEstadísticas resumidasEstadísticas por ejeDifusiónMáscaras booleanasRepaso: calificaciones de exámenes6Gráficos para datos
Gráficos de barrasHistogramasVarios gráficos en unoImágenes como matricesRepaso: informe de precipitaciones9Agrupar y limpiar
Contar valoresAgrupar datosVarios resúmenesValores faltantesRepaso: resultados de la encuestaPractica por tu cuenta: Compilador de Python online