分类：axis=0

2021
08-06

python之np.argmax()及对axis=0或者1的理解

对于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)这个问题可以帮助我们理解... 继续阅读 >