Discussion :

[ludovic1975]

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 :

Sélectionner tout - Visualiser dans une fenêtre à part

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

[ludovic1975]

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 :

Sélectionner tout - Visualiser dans une fenêtre à part

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 :

Sélectionner tout - Visualiser dans une fenêtre à part

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