Как получить среднее значение трехмерного массива над 2D?

Question

У меня есть массив со следующей формой (7352, 128, 6)

Как я могу получить "среднее" для 1-го и 2-го измерений с NumPy в Python?

Полученная форма "средства", которую я хочу получить, равна (1, 6).

Answer 1

Вы можете сделать:

np.mean(x, axis=(0, 1))

Ваша полученная фигура будет (6,), хотя.

Answer 2

Вы можете использовать параметр оси функции mean :

img = np.random.rand(7352,128,6)
y_mean = img.mean(axis=0) # mean over y (row)
x_mean = img.mean(axis=1) # mean over x (column)
z_mean = img.mean(axis=(0,1)) # the mean you want (shape is (6,))

Эта запись также может помочь.

Answer 3

numpy.mean() с аргументом оси является правильным решением.

import numpy as np

x = np.random.rand(7352, 128, 6)
x_mean = np.mean(x, axis=(0, 1))

print(x_mean.shape)  # -> (6,)