分类：np.argmin函数

2021
08-06

Python中的np.argmin()和np.argmax()函数用法

Pythonnp.argmin()和np.argmax()函数按照axis的要求返回最小的数/最大的数的下标numpy.argmin(a,axis=None,out=None)numpy.argmax(a,axis=None,out=None)a：传入一个数组，axis：默认将输入数组展平，否则，按照axis方向out：可选importnumpyasnpa=np.arange(6).reshape(2,3)aarray([[0,1,2],[3,4,5]])#此时自动展平了np.argmin(a)0np.argmax(a)5#在axis方向上找最小的值并返回坐标np.argmin(a,axis=1)ar... 继续阅读 >