怎么在python中使用np.argmax() 返回最大值索引号
怎么在python中使用np.argmax() 返回最大值索引号?很多新手对此不是很清楚,为了帮助大家解决这个难题,下面小编将为大家详细讲解,有这方面需求的人可以来学习下,希望你能有所收获。
import numpy as np a = np.array([[2, 4, 6, 1], [1, 5, 2, 9]]) print(np.argmax(a)) print(np.argmax(a, axis=0)) #竖着比较,返回行号 print(np.argmax(a, axis=1)) #横着比较,返回列号
输出:
7
[0 1 0 1]
[2 3]
Python np.argmax()的用法
用法:求的是最大值对应的索引
案例如下
看完上述内容是否对您有帮助呢?如果还想对相关知识有进一步的了解或阅读更多相关文章,请关注行业资讯频道,感谢您对亿速云的支持。