Вычислить среднее из вспомогательных массивов в пределах большей 3D массива в Python

Question

У меня есть 3D-массив так:

a = ([[[25, 4, 54], [58, 14, 27], [35, 40, 62], [53, 72, 86], [87, 4, 51], [89, 34, 22]], 
     [[61, 99, 89], [52, 93, 18], [ 0, 38, 12], [99, 10, 98], [79, 91, 75], [31, 59, 3]], 
     [[22, 9, 63], [37, 26, 3], [54, 40, 39], [36, 87, 44], [81, 10, 7], [75, 29, 20]], 
     [[55, 30, 44], [21, 36, 98], [56, 52, 78], [80, 76, 32], [72, 26, 17], [15, 81, 37]], 
     [[73, 37, 28], [30, 44, 75], [66, 59, 61], [34, 57, 56], [13, 51, 73], [20, 51, 94]], 
     [[86, 38, 69], [85, 74, 91], [25, 90, 58], [50, 12, 79], [24, 96, 28], [73, 2, 17]]]) 

и я пытаюсь вычислить среднее из 4 элементов каждой последующих строки и столбца, затем , по всей матрице, чтобы имели средние квадраты средних квадратов большой матрицы. Я ожидаю иметь в результате целочисленная матрица, как:

([[[49, 52, 47], [47, 40, 64], 

... и так далее ...

Я попытался с помощью нарезку с шагом следуют средней функции как

a[::2].mean() 

, но он не возвращает результат, который я ожидаю.

Пока что я могу получить это среднее столбца или строки с помощью

numpy.mean(a, axis=0 or 1) 

, но это не полезно для меня, потому что она возвращает весь столбец или строки означают, а не средство 4 элементов через матрицу.

Каков наилучший способ для этого?

Answer 1

Я предполагаю, что вы хотите вычислить среднее значение записей (0,0), (0,1), (1,0), (1,1), затем (2,2), (2 , 3), (3,2), (3,3) и т. Д.

Вы выбираете каждый второй столбец и строку четыре раза, начиная с индексов (0,0), (0,1), (1,0) и (1,1) соответственно. Затем разделить на 4 и округлить результат:

np.round((a[::2,::2] + a[1::2,::2] + a[::2,1::2] + a[1::2,1::2])/4.0) 

Это дает

array([[[ 49., 52., 47.], 
     [ 47., 40., 64.], 
     [ 72., 47., 38.]], 

     [[ 34., 25., 52.], 
     [ 56., 64., 48.], 
     [ 61., 36., 20.]], 

     [[ 68., 48., 66.], 
     [ 44., 54., 64.], 
     [ 32., 50., 53.]]])