Python中np.argmax()函数用法示例
前言
np.argmax是用于取得数组中每一行或者每一列的的最大值。常用于机器学习中获取分类结果、计算精确度等。
函数：numpy.argmax(array, axis)
array：代表输入数组；axis：代表对array取行（axis=0）或列（axis=1）的最大值。
一、一维数组的用法
x = np.arange(12) # [ 0 1 2 3 4 5 6 7 8 9 10 11]
index = np.argmax(x)
print("1 dimension test:",index) # 11运算后，降一维，成为一个数值，11的索引值维11，所以运算结果为：11
二、二维数组的用法
# 0代表对行进行最大值选取，此时对每一列进行操作
x = np.arange(12).reshape(3,4)
print(x)
index = np.argmax(x,axis = 0)
print("2 dimension test, axis = 0:",index)
# 输出结果：
# x：[[ 0 1 2 3]
# [ 4 5 6 7]
# [ 8 9 10 11]]
# 2 dimension test, axis = 0:[2 2 2 2]
# 1代表对列进行最大值选取，此时对每一行进行操作
x = np.arange(12).reshape(3,4)
index = np.argmax(x,axis = 1)
print("2 dimension test, axis = 1:",index)
# 输出结果：
# 2 dimension test, axis = 1:[3 3 3]（1）遵循运算之后降一维的原则，因此返回的会是一个一维的array。
（2）函数返回的是最大值的索引，而不是最大值本身。
（3）另一种理解思路：0：外层；1：内层。axis的取值为0和1，剥掉对应的中括号，将里面的内容直接按逗号分隔。
例：x=[[ 0 1 2 3][ 4 5 6 7][ 8 9 10 11]]是一个（3,4）的矩阵，当axis=0时，剥掉3，返回一个（1,3）的矩阵；在axis=1时，剥掉4，返回一个（1,3）的矩阵。
析：当axis=0时，剥掉外层---->[0 1 2 3]
[4 5 6 7 ]
[8 9 10 11]
[argmax(0,4,8), argmax(1,5,9), argmax(2,6,10),argmax(3,7,11)] # 输出：[2,2,2,2]
当axis=1时，剥掉内层---->[0 1 2 3
4 5 6 7
8 9 10 11]
[argmax(0,1,2,3), argmax(4,5,6,7), argmax(8,9,10,11)], # 输出：[3,3,3]
三、三维数组的用法
一个m×n×p维的矩阵：
- axis为0，舍去m，返回一个 n×p 维的矩阵
- axis为1，舍去n，返回一个 m×p 维的矩阵
- axis为2，舍去p，返回一个 m×n 维的矩阵
# 三维向量测试
# 0代表进入第一个[]内进行对比
x = np.arange(24).reshape(2,3,4)
x[1,0,3] = 1
# x =
# [[[ 0 1 2 3]
# [ 4 5 6 7]
# [ 8 9 10 11]]
# [[12 13 14 1]
# [16 17 18 19]
# [20 21 22 23]]]
index = np.argmax(x,axis = 0)
print("3 dimension test, axis = 0:",index)
# 当axis=0时，进入第一个[]内进行对比，此时x剩下两部分。
# [[ 0 1 2 3]
# [ 4 5 6 7]
# [ 8 9 10 11]]
# [[12 13 14 1]
# [16 17 18 19]
# [20 21 22 23]]
# 两部分格式相同，将剩下的两部分每一个单位进行对比，对比结果为
# [[1 1 1 0]
# [1 1 1 1]
# [1 1 1 1]]
# 除去我设置的特殊位置外，其他位置均为第二部分大。
# 三维向量测试
# 1代表进入第二个[]内进行对比
# x =
# [[[ 0 1 2 3]
# [ 4 5 6 7]
# [ 8 9 10 11]]
# [[12 13 14 1]
# [16 17 18 19]
# [20 21 22 23]]]
index = np.argmax(x,axis = 1)
print("3 dimension test, axis = 1:",index)
# 当axis=1时，进入第二个[]内进行对比。
# [ [ 0 1 2 3]
# [ 4 5 6 7]
# [ 8 9 10 11]
# [12 13 14 1]
# [16 17 18 19]
# [20 21 22 23] ]
# 对于第二个[]内的内容而言，均剩下三部分，我特意将两个第二个[]内的内容分开更容易辨认
# 第一个是
# [ 0 1 2 3]
# [ 4 5 6 7]
# [ 8 9 10 11]
# 第二个是
# [12 13 14 1]
# [16 17 18 19]
# [20 21 22 23]
# 都是第三行的值最大，所以输出结果为
# [[ 2 2 2 2]
# [ 2 2 2 2]]
# 三维向量测试
# 2代表进入第三个[]内进行对比
x = np.arange(24).reshape(2,3,4)
x[1,0,3] = 1
# x =
# [[[ 0 1 2 3]
# [ 4 5 6 7]
# [ 8 9 10 11]]
# [[12 13 14 1]
# [16 17 18 19]
# [20 21 22 23]]]
index = np.argmax(x,axis = 2)
print("3 dimension test, axis = 2:",index)
# 当axis=2时，进入第三个[]内进行对比。
# [[ 0 1 2 3
# 4 5 6 7
# 8 9 10 11 ]
# [ 12 13 14 1
# 16 17 18 19
# 20 21 22 23 ]]
# 对于第三个[]内的内容而言，均剩下四部分，我特意将六个第三个[]内的内容分开更容易辨认
# 第一个是
# 0 1 2 3
# 第二个是
# 4 5 6 7
# ……
# 最后对比结果为
# [[ 3 3 3 ]
# [ 2 3 3 ]]参考：
总结
到此这篇关于Python中np.argmax()函数用法的文章就介绍到这了,更多相关Python np.argmax()函数内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
这篇文章主要介绍了详解Python发送email的三种方式，Python发送email的三种方式，分别为使用登录邮件服务器、使用smtp服务、调用sendmail命令来发送三种方法，非常具有实用价值，需要的朋友可以参考下2018-10-10
解决pandas无法在pycharm中使用plot()方法显示图像的问题
今天小编就为大家分享一篇解决pandas无法在pycharm中使用plot()方法显示图像的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-05-05
这篇文章主要介绍了将python打包后的exe还原成py,利用pyinstxtractor.py 拆包（解压）工具，将exe文件解压成一个文件夹<BR>uncompyle6 pyc反编译工具,需要的朋友可以参考一下2022-01-01
将图片翻译成文字一般被称为光学文字识别（Optical Character Recognition,OCR）,这篇文章主要给大家介绍了关于Python实现OCR识别的相关资料,文中通过示例代码介绍的非常详细,需要的朋友可以参考下2021-08-08
今天小编就为大家分享一篇PyCharm在新窗口打开项目的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-01-01
这篇文章主要介绍了Python3实现获取图片文字里中文的方法,结合实例形式分析了Python基于文字识别引擎tesseract-ocr针对图片里中文识别的相关操作技巧与操作注意事项,需要的朋友可以参考下2018-12-12
本文介绍Python-Selenium自动化爬虫,Selenium是一个Web的自动化测试工具，最初是为网站自动化测试而开发的，Selenium 可以直接运行在浏览器上,它支持所有主流的浏览器,可以接收指令,让浏览器自动加载页面,获取需要的数据,甚至页面截屏,xiamian neir 需要的朋友可以参考下2022-01-01
plt.imshow()是Matplotlib库中的一个函数,主要用于显示图像或矩阵数据,本文主要介绍了Python plt.imshow函数及其参数使用,具有一定的参考价值,感兴趣的可以了解一下2024-02-02
这篇文章主要介绍了利用一个简单的例子窥探CPython内核的运行机制,作者通过一个简单的输出函数深入、介绍了CPython源码C代码中的一些函数，需要的朋友可以参考下2015-03-03
这篇文章主要介绍了基于python3抓取pinpoint应用信息入库,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-01-01
最新评论