web-dev-qa-db-fra.com

Moyenne sur tous les n éléments d'un tableau numpy

J'ai un tableau numpy. Je veux créer un nouveau tableau qui est la moyenne de chaque triplet consécutif d'éléments. Ainsi, le nouveau tableau sera un tiers de la taille de l'original.

Par exemple:

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

devrait retourner le tableau:

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

Quelqu'un peut-il suggérer un moyen efficace de le faire? Je dessine des blancs.

43
user1654183

Si votre tableau arr a une longueur divisible par 3:

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

Le remodelage vers un tableau de dimensions supérieures, puis l'exécution d'une certaine forme d'opération de réduction sur l'une des dimensions supplémentaires est un élément essentiel de la programmation numpy.

72
Jaime