应该如何打印矩阵的所有索引，这些索引是该矩阵的最大数量？

Question

就像我们可以考虑以下矩阵： enter image description here

现在我想显示矩阵最大值的所有索引，而不仅仅是矩阵最大值的一个索引。

Answer 1

import numpy as np

mat = np.array([[2,3,2], [7,7,6], [2,7,3]])
print(mat)

max_indices = np.where(mat == np.amax(mat))
print(max_indices)

index_max = mat[max_indices]
print(index_max)

输出：

[[2 3 2]
 [7 7 6]
 [2 7 3]]
(array([1, 1, 2]), array([0, 1, 1]))  # first array: x-axis, second: y-axis
[7 7 7]