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

Posted 2022-03-08 卡尔曼和玻尔兹曼谁曼

Python特别灵活，肯定方法不止一种，这里介绍一种我觉得比较简单的方法。
如下图，使用x == np.max(x) 获得一个掩模矩阵，然后使用where方法即可返回最大值对应的行和列。
where返回一个长度为2的元组，第一个元素保存的是行号，第二个元素保存的是列号。