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

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

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