Поиск максимальных значений из каждого подмассива массива numpy ⇐ Python

[Anonymous]

У меня есть массив:

Код: Выделить всё

x = array([[[ 0,  1,  2],
[ 6,  7,  8],
[12, 13, 14]],

[[ 3,  4,  5],
[ 9, 10, 11],
[15, 16, 17]],

[[18, 19, 20],
[24, 25, 26],
[30, 31, 32]],

[[21, 22, 23],
[27, 28, 29],
[33, 34, 35]]])

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

Код: Выделить всё

output = array([14,17,32,35])

Теперь это можно легко сделать с помощью цикла, однако я хочу этого избежать. np.max(x) выдает результат 35, то есть максимальное значение всего массива. np.max(axis) тоже не работает (я не уверен, что это сработает, но я все равно попробовал)
Кто-нибудь, можете помочь?


Подробнее здесь: https://stackoverflow.com/questions/674 ... umpy-array