np.argmax()函数的作用

版权声明:创作不易,转载请留出处,博客如有错误,欢迎评论帮改正。 https://blog.csdn.net/qq_39072607/article/details/89363950

np.argmax函数的作用

np.argmax函数参数

argmax(a, axis=None, out=None)
返回axis维度的最大值的索引
a :输入一个array类型的数组。
axis:参数为None时默认比较整个数组,参数为0按列比较,参数为1按行比较。
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/qq_39072607/article/details/89363950