python - transpuesta - Un promedio de cada n elementos de una matriz numpy

sumar columnas de una matriz python

Question

Tengo una matriz numpy. Quiero crear una nueva matriz que sea el promedio de cada tercer elemento. Así que la nueva matriz será un tercio del tamaño como el original.

Como ejemplo:

 np.array([1,2,3,1,2,3,1,2,3])

debe devolver la matriz:

 np.array([2,2,2])

¿Alguien puede sugerir una manera eficiente de hacer esto? Estoy dibujando espacios en blanco.

Answer 1

Si su matriz arr tiene una longitud divisible por 3:

np.mean(arr.reshape(-1, 3), axis=1)

La remodelación a una matriz dimensional superior y luego realizar algún tipo de operación de reducción en una de las dimensiones adicionales es un elemento básico de la programación de números.