当前位置:主页 > 软件编程 > Python代码 >

Python 实现Numpy中找出array中最大值所对应的行和列

时间:2021-02-03 18:26:43 | 栏目:Python代码 | 点击:

Python特别灵活,肯定方法不止一种,这里介绍一种我觉得比较简单的方法。

如下图,使用x == np.max(x) 获得一个掩模矩阵,然后使用where方法即可返回最大值对应的行和列。

where返回一个长度为2的元组,第一个元素保存的是行号,第二个元素保存的是列号。

您可能感兴趣的文章:

相关文章