Arrays 索引numpy矩阵的最高值

我有一个形状（4，7）的numpy数组，如下所示：

array([[ 1,  4,  5,  7,  8,  6,  7]
       [ 2, 23,  2,  4,  8, 94,  2],
       [ 1,  5,  6,  7, 10, 15, 20],
       [ 3,  9,  2,  7,  6,  5,  4]])

---

我想得到最高元素的索引，即94，格式如下：第一行第五列。因此，输出应该是一个numpy数组（[1,5]）（matlab风格）。

您可以使用获得最大索引的索引，但要获得实际的行和列，必须使用以下方法：

import numpy as np

arr = np.array([[ 1,  4,  5,  7,  8,  6,  7],
                [ 2, 23,  2,  4,  8, 94,  2],
                [ 1,  5,  6,  7, 10, 15, 20],
                [ 3,  9,  2,  7,  6,  5,  4]])

maximum = np.unravel_index(arr.argmax(), arr.shape)

print(maximum)
# (1, 5)

您必须使用

np.unlavel_index

，因为默认情况下

np.argmax

将从展平数组返回索引（在您的情况下是索引12）