2017-06-15 54 views

回答

0

使用amax功能與NumPy,並沿着指定要採取的最大軸:

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

感謝您的答覆。 amax不會給我我想要的,因爲我想要最大值的z-index。不過,我從那以後發現了np.argmax,它給了我這個。唯一的問題是它不會忽略nan,當它找到它時,它將返回其索引作爲最大值的位置。我可能可以將所有的nan變成一個非常小的數字,然後運行np.argmax,但似乎應該採取更少的方法來做到這一點。 – Stacy

相關問題