沿Numpy ndarray轴的最大值？

Question

numpy_array http://www.freeimagehosting.net/newuploads/klipa.jpg

我担心我无法描述问题所以我画了一个草图。无论如何，我需要的是在numpy ndarray中找到沿着第0轴的最大值，iearray.shape（5， 5,3），以及它们对应的“层数”，并使用“层数”创建一个形状为（1,5,3）的新的2d数组。希望我在这里给出一个清晰的描述..谢谢很多。

Answer 1

如果查看np.max的文档，您会看到它需要axis个参数：

a.max(axis=0)

但那对你没有帮助。但是，有一个函数argmax可以给出给定轴上最大值的索引：

a.argmax(axis=...)

所以，让我们找到你的第一个（5,5）数组：它是a[...,0]。您可以使用a[...,0].max(axis=1)（或0）找到每行（或列）的最大值的位置，并使用它来查找另一侧的值。