8
有一種方法可以一次獲得max和argmax嗎?Numpy argmax。如何計算max和argmax?
import numpy as np
a=[0,0,1,0]
maximum=max(a)
index=np.argmax(a)
有沒有做到這一點,與像一個最快的方法:
[maximum,index]=function(a)
有一種方法可以一次獲得max和argmax嗎?Numpy argmax。如何計算max和argmax?
import numpy as np
a=[0,0,1,0]
maximum=max(a)
index=np.argmax(a)
有沒有做到這一點,與像一個最快的方法:
[maximum,index]=function(a)
也許這樣的事情是快...
index=np.argmax(a)
max=a(index)
上節省計算時間好一點,然而,如果最初的argmax調用是在第二個軸上對二維數組進行排序,那麼第二個語句可能會引發錯誤。 – PaulDong 2016-12-23 06:34:37
你可以使用unravel_index來恢復它(https://docs.scipy.org/doc/numpy/reference/generated/numpy.unravel_index.html) – PaxRomana99 2017-01-27 18:11:11