Nump: использование np.mean в 3-х измерениях

Question

У меня есть 140 массивов, размер которых равен 734 x 1468. Мне нужно среднее значение по 140 массивам в записях 734 x 1468, я начал с итерирования данных с помощью вложенных циклов но это слишком медленно. Я попытался с помощью функции np.mean

res = np.mean([megadatalist[i].variables['analysed_sst'][0] for i in range(0,140)], axis=2) 

res.shape дает (140, 734)

Мне нужен результат измерений 734, 1468

megadatalist состоит из 140 NetCDF файлов.

Answer 1

Вы передаете список массивов, на которые делается ссылка (140,734,1468). Поэтому либо вы используете np.dstack, как показано ниже, либо можете изменить axis=0.

С np.dstack:

X = np.dstack([megadatalist[i].variables['analysed_sst'][0] 
     for i in range(0,140)]) 

res = np.mean(X, axis=2)