分类:Python
2021
08-06
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...
继续阅读 >
对于np.argmax()让我迷惑了很久,尤其是其中的axis=1的比较结果。一、np.argmax()的理解1、最简单的例子假定现在有一个数组a=[3,1,2,4,6,1]现在要算数组a中最大数的索引是多少。最直接的思路,先假定第0个数最大,然后拿这个和后面的数比,找到大的就更新索引。代码如下a=[3,1,2,4,6,1]maxindex=0i=0fortmpina:iftmp>a[maxindex]:maxindex=ii+=1print(maxindex)这个问题可以帮助我们理解...
继续阅读 >
2021
08-06