在 Python 中最大化 4D 数组

maximizing over a 4D array in Python

所以我在 Python 中有一个 x b x c x d 数组。称其为 S。我想创建一个数组 a x b,将其称为 V,以便 V[i,j] 等于 S[i,j,., .] 视为 c x d 数组所采用的最高值。我也想知道坐标(大约0

使用 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))