numpy réduire d'une dimension après sommation des valeurs la composant

Bonjour,

Avec numy, je souhaite réduire mon array qui a un shape de (240,320,3) à un shape de (240,320) en ayant au passage sommé la 3° dimension.
Avec un for on y arrive très bien mais ca prend trop de temps. J'imagine que numpy peut faire ca autrement mais je sèche un peu...

Donc comment remplacer le code suivant (avec a.shape = (240,320,3)

Code:

---

1 2 3 4 5

b = np.zeros(shape=(240,320)).astype(np.int) for i in range(240): for j in range(320): b[i,j] = a[i,j,...].sum()

---

Par un code plus rapide à l'execution ?

Si vous avez une piste je suis preneur...

bonne fin de semaine
Ludovic

Bon en approfondissant la doc numpy https://docs.scipy.org/doc/numpy/ref...y.ufunc.reduce

J'ai trouvé mon bonheur
Je remplacer le code

Code:

---

1 2 3 4 5

b = np.zeros(shape=(240,320)).astype(np.int) for i in range(240): for j in range(320): b[i,j] = a[i,j,...].sum()

---

par celui là

Code:

---

1 2	b = np.add.reduce(a, axis=2,keepdims=False)

---

Je gagne 3 secondes (sur mon script entier : 4,8s pour la boucle for, contre 1,8s pour le second code).

Je peux me coucher maintenant...
Ludovic