Как получить максимум 3D-матрицы на основе первого элемента в python

Question

У меня есть трехмерная матрица с формой (3,2,2). Как это:

t[0]=[2,4 
     5,6]  

t[1]=[3,3 
     2,3] 

t[2]=[1,5 
    4,7] 

Я хочу, чтобы получить максимум t, чтобы дать мне 6, 3 и 7 что 6 является макс t[0] и так далее.

Answer 1

Один из вариантов: Вы можете перебрать массив и принять максимальное значение для каждого подпроекта массива:

[np.max(x) for x in t] 
# [6, 3, 7] 

пыльников вариант:

t.reshape(3,4).max(axis = 1) 
# array([6, 3, 7] 

Более простой метод является:

t.max(axis = (1,2)) 
# array([6, 3, 7]) 

---

Немного о знаке вышеупомянутые три метода:

%timeit [np.max(x) for x in t] 
# 100000 loops, best of 3: 10.9 µs per loop 

%timeit t.reshape(3,4).max(axis = 1) 
# 100000 loops, best of 3: 2.75 µs per loop 

%timeit t.max(axis = (1,2)) 
# 100000 loops, best of 3: 2.63 µs per loop