求一个数组中某个值（最大值）索引的方法

xiaofei-liu

于 2020-01-03 16:03:03 发布

阅读量2.3k

点赞数

分类专栏： python基础

本文链接：https://blog.youkuaiyun.com/qq_35690171/article/details/103822009

版权

python基础专栏收录该内容

12 篇文章

订阅专栏

本文介绍了两种在数组中找到最大值索引的方法，一种能返回第一个最大值的索引，但可能无法处理存在多个最大值的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

两种方法：

#method1
np.argwhere(a == np.max(a))

#method2
np.unravel_index(a.argmax(), a.shape)

此方法不是很好，如果有多个最大值，只能返回第一个
展示：
在这里插入图片描述

code:

import numpy as np
a = np.array([[2, 4, 6, 1], [1, 5, 2, 9],[3, 6, 8, 7]])
a
print(np.argmax(a))
print(np.argmax(a, axis=0))  #竖着比较，返回行号
print(np.argmax(a, axis=1))  #横着比较，返回列号

np.unravel_index(a.argmax(), a.shape)

np.argwhere(a == np.max(a))