python-3.x - 在网格中查找最大值的坐标

Question

给定一个像这样的数字网格

g=[[2, 2, 2, 3, 2],
   [2, -1, -1, 3, 3],
   [3, 2, -1, 3, 3],
   [2, 4, 3, 3, 3],
   [-1, 2, 2, -1, 1]]

我正在那个 2d 列表中搜索最大值，在这个例子中，显然是 4。为了找到这一点， max(max(graph,key=max)) 完成了这项工作(尽管我知道它两次调用最大值为最大值的行，所以它不是非常有效)

但这只是我正在寻找的一半。我需要这个 4 的坐标。我想要一段简短、高效且像 Python 一样的代码x, y = find_2d_max_index(g) 这给了我最大数的位置，而之前没有做三重最大的事情。我想避免在找到最大值后使用 index() 方法，因为那样效率很低。

我查看了 numpy argmax，但无法让它在这个带有 2d 列表的应用程序中工作。

感谢任何帮助。

Answer 1

您可以为此使用 numpy 和 documentation说明如何。

import numpy as np

a = np.array([[2, 2, 2, 3, 2],
   [2, -1, -1, 3, 3],
   [3, 2, -1, 3, 3],
   [2, 4, 3, 3, 3],
   [-1, 2, 2, -1, 1]])

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

a[idx]
>> 4

说明:如果数组是一个平面数组，我们使用未定义轴的 argmax 来获取最大项的索引，在这种情况下它将是 16 为数字 4 将是这个长数组的 16th 元素。然后我们将实际数组的形状提供给 np.unravel_index 和最大项的索引，在本例中为 16 并根据数组的形状确定，这个数字的坐标。