numpy - 沿 Numpy ndarray 轴的最大值？

Question

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

Answer 1

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

a.max(axis=0)

但这还不能帮助你。但是，有一个函数argmax可以为您提供沿给定轴的最大值的索引：

a.argmax(axis=...)

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