沿Numpy ndarray轴的最大值？

---

恐怕我无法描述这个问题，所以我画了一个草图。无论如何，我需要的是在一个numpy数组中找到沿第0轴的最大值，即array.shape（5,5,3）及其相应的“层号”，并使用“层号”创建一个形状为（1,5,3）的新二维数组.希望我在这里给出一个清晰的描述..非常感谢。

如果您查看

np.max

的文档，您将看到它包含一个

轴

参数：

a.max(axis=0)

但这对你还没有帮助。但是，有一个函数

argmax

，它提供了沿给定轴的最大值索引：

a.argmax(axis=...)

那么，让我们找到您的第一个（5,5）数组：它是

a[…，0]

。您可以使用

a[…，0].max（axis=1）

（或0）查找每行（或每列）最大值的位置，并使用该位置查找其他面上的值。

如何获取最大值的“层号”？谢谢！实际上，argmax（axis=0）返回我需要的数组