Wie berechne ich den Mittelwert eines 4D Arrays?

Question

Ich habe ein Array img mit Form ist 64x128x512x3, die von drei Bildern 64x128x512 concated ist. Ich möchte den Mittelwert jedes Bildes einzeln berechnen, gegeben durch das Array img. Daher führte ich den Code als Faltenbalg:

import numpy as np 
img_means = np.mean(img, (0, 1, 2))) 

Ist es korrekt? Mein erwartetes Ergebnis ist, dass img_means[0,:,:,:] Mittel des ersten Bildes ist, img_means[1,:,:,:] ist das Mittel des zweiten Bildes, img_means[2,:,:,:] des dritten Bildes.

Answer 1

Ja, es ist korrekt, aber beachte, dass img_means nur ein Array von drei Zahlen ist (jede ist der Mittelwert der entsprechenden Zahl).

Answer 2

Ihr Code funktioniert nicht in Python 3.x es wie folgt tun: zuerst die Daten

import numpy as np 
img=np.arange(64*128*512*3).reshape(64,128,512,3) 

erzeugen Und das ist, was Sie wollen:

img_means=[img[:,:,:,i].mean() for i in range(img.shape[3]) ]