np.argmax的注意点

np.argmax函数

当参数是2维时，

axis=-1，代表纵轴，即选择纵轴方向最大的元素下标（即每一行的最大值）

axis=0，代表横轴，即选择横轴方向最大的元素下标（即每一列的最大值）

 

多维时，以此类推。

 

更多学习笔记可以关注我的微信公众号「kelly学挖掘」，欢迎交流。

-- over --

版权声明：本文为qm5132原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。