如何在一个numpy对象数组中找到最小/最大对象的索引？

import numpy as np

arr = np.array([5, 2, 9, 1, 7])
min_index = np.argmin(arr)
print(min_index)  # 输出: 3

import numpy as np

arr = np.array([5, 2, 9, 1, 7])
max_index = np.argmax(arr)
print(max_index)  # 输出: 2

import numpy as np

arr = np.array([[5, 2, 9], [1, 7, 3]])
min_index = np.argmin(arr, axis=1)
max_index = np.argmax(arr, axis=0)

print(min_index)  # 输出: [1 0]
print(max_index)  # 输出: [0 1 0]