找到numpy 3d数组沿z轴的最大索引

时间:2017-06-15 23:39:33

标签: python arrays numpy

我有一个3d numpy数组。我想在阵列的z轴上找到非零元素的每个[x,y]网格点处的最大坐标。这将导致2d数组填充z索引值。我怎么能这样做?

1 个答案:

答案 0 :(得分:0)

使用Numpy中的amax功能,并指定您想要获取最大值的轴:

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