n.median () возвращает 1d скаляр для массива глубины 3?

Question

У меня есть двумерный массив.Каждая ячейка / элемент в массиве содержит скаляр / кортеж, представляющий цвет RGB.Когда я пытаюсь вычислить numpy.median() или numpy.average() в массиве, я получаю 1-й скаляр, когда мне нужно вернуть 3D-скаляр / кортеж.

Например;

f = (1,2,3)
a = np.array([[f,f,f],
    [f,f,f]])
print("A: {}, {}".format(np.median(a), a.shape))

Вывод:

A: 2.0, (2, 3, 3)

Разве среднее значение не должно быть (1,2,3), а не 2.0?

Answer 1

Вы хотите, чтобы np.median(a, (0,1)) вычислило медиану по первым двум осям.