Python 沿numpy 3d阵列的z轴查找最大索引

我有一个3d numpy阵列。我想在阵列z轴上非零元素的每个[x，y]网格点上找到最大坐标。这将导致二维数组中填充z索引值。如何执行此操作？

使用Numpy中的

amax

功能，并指定要获取最大值的轴：

ans = numpy.amax(arr_3D, axis=2)

---

谢谢你的回复。amax没有给出我想要的，因为我想要最大值的z索引。然而，我已经找到了np.argmax，它给了我这一点。唯一的问题是它不会忽略nan，当它找到一个nan时，它会返回索引作为最大值的位置。我可能会将所有nan都更改为一个非常小的数字，然后运行np.argmax，但似乎应该有一种不那么麻烦的方法来实现这一点。如果其他人遇到这种情况，您可能需要

np.nanargmax

。