0
我有一個3d numpy數組。我想找到沿着數組z軸的非零元素的每個[x,y]網格點的最大座標。這將導致用z索引值填充的2d數組。我怎樣才能做到這一點?沿着numpy 3D陣列的z軸查找最大指數
我有一個3d numpy數組。我想找到沿着數組z軸的非零元素的每個[x,y]網格點的最大座標。這將導致用z索引值填充的2d數組。我怎樣才能做到這一點?沿着numpy 3D陣列的z軸查找最大指數
使用amax
功能與NumPy,並沿着指定要採取的最大軸:
ans = numpy.amax(arr_3D, axis=2)
感謝您的答覆。 amax不會給我我想要的,因爲我想要最大值的z-index。不過,我從那以後發現了np.argmax,它給了我這個。唯一的問題是它不會忽略nan,當它找到它時,它將返回其索引作爲最大值的位置。我可能可以將所有的nan變成一個非常小的數字,然後運行np.argmax,但似乎應該採取更少的方法來做到這一點。 – Stacy