在Python中最大化4D数组

Question

所以我在Python中有一个x b x c x d数组。将其称为S.我想创建一个数组a x b，将其称为V，使得V [i，j]等于S [i，j，。，。]看作c x d数组的最高值。我也想知道坐标（大约0

Answer 1

使用numpy沿轴获取最大值。 axis参数指的是您最大限度地采用的维度。

import numpy as np

a,b,c,d = 3,4,5,6 #Whatever you want
S = np.random.random( (a,b,c,d))
V = np.max(S.reshape( (a,b,c*d) ), axis = 2)
#OR
V = np.max(np.max(S,axis = 3),axis = 2)
#OR
V = np.max(S, axis = (2,3))