Python Numpy确定包含总最大/最小值的数组索引

import numpy as np

# 创建一个二维数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 使用argmax函数确定包含总最大值的索引
max_index = np.argmax(arr)
# 使用unravel_index函数将一维索引转换为二维索引
max_index = np.unravel_index(max_index, arr.shape)

# 使用argmin函数确定包含总最小值的索引
min_index = np.argmin(arr)
# 使用unravel_index函数将一维索引转换为二维索引
min_index = np.unravel_index(min_index, arr.shape)

print("总最大值的索引：", max_index)
print("总最小值的索引：", min_index)

总最大值的索引： (2, 2)
总最小值的索引： (0, 0)