沿着numpy 3D阵列的z轴查找最大指数

我有一个3d numpy数组。我想找到沿着数组z轴的非零元素的每个[x，y]网格点的最大坐标。这将导致用z索引值填充的2d数组。我怎样才能做到这一点？沿着numpy 3D阵列的z轴查找最大指数

2017-06-15 Stacy

使用amax功能与NumPy，并沿着指定要采取的最大轴：

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

来源

2017-07-26 04:02:32 2cynykyl

感谢您的答复。 amax不会给我我想要的，因为我想要最大值的z-index。不过，我从那以后发现了np.argmax，它给了我这个。唯一的问题是它不会忽略nan，当它找到它时，它将返回其索引作为最大值的位置。我可能可以将所有的nan变成一个非常小的数字，然后运行np.argmax，但似乎应该采取更少的方法来做到这一点。 – Stacy

沿着numpy 3D阵列的z轴查找最大指数

回答

相关问题