Agregación por ejes en Python

Reducir arreglos con mean, sum y compañía por eje.

import numpy as np

puntajes = np.array([[85, 90, 78], [72, 88, 95], [60, 70, 80]])

general = puntajes.mean()
por_alumno = puntajes.mean(axis=1)
por_prueba = puntajes.max(axis=0)
dispersion = puntajes.std()

total = puntajes.sum()
mejor_indice = puntajes.argmax()
acumulado = puntajes.cumsum(axis=1)

Cómo funciona

1. Sin eje se reduce el arreglo completo.
2. axis=1 colapsa columnas; axis=0 colapsa filas.
3. argmax y cumsum localizan y acumulan.

Palabras clave y builtins usados aquí

• as

El intento, en números

Líneas

12

Caracteres a escribir

300

Tokens

94

Ritmo de tres estrellas

105 tpm

Al ritmo de tres estrellas de 105 tokens por minuto, este intento toma unos 54 segundos.

Escribe este fragmento

Paso 2 de 4 en Matemática y formas; paso 4 de 22 en NumPy.

← Anterior Siguiente →