1 问题描述

最近有一个需求就是寻找矩阵最大值对应的索引，然后取出最大值对应的行或者列。最核心的需求就是如何找到最大值对应的索引。在网络上搜索了很多方法，都不是特别清楚，自行去Numpy库寻找，发现了很好用的办法np.argwhere。因此很多问题大家还是需要去官方库多寻找解决方案，百度的结果不一定有你所需要的。

2 代码说明

2.1 最大值对应的索引

代码截图如下：
首先生成我们需要的矩阵

这里我们使用的矩阵A和B如下

随后我们使用np.argwhere(A==A.max())来寻找最大值对应的索引

输出如下：

2.2 寻找某个特定值对应的索引

如果需要寻找某个特定值比如：在B矩阵中寻找元素2对应的索引，可以使用：np.argwhere(B==2)

我们可以得到如下结果

参考文献

[1] https://numpy.org/doc/stable/reference/generated/numpy.argwhere.html