python - 获取 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]

标签： pythonnumpy

Answer 1

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

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