获取3d numpy的最大值

Question

假设给定以下3d numpy：

myArray = np.array([[[1,1,5],
[1,1,1],
[1,6,1]],
[[2,2,2],
[2,3,2],
[2,2,2]],
[[3,3,1],
[7,1,1],
[1,3,3]]])

需要计算行的最大值并返回长度为3的列表。

即，在上面的例子中，预期的结果是：

[7,6,5]

Answer 1

首先，您需要减少一个轴，然后您可以使用np.max

myArray = myArray.ravel().reshape(9,3)
np.max(myArray,axis=0)