numpy获取数组最大值及其下标

可以使用numpy的argmax函数来获取数组的最大值及其下标。例如，对于一个一维数组a：

import numpy as np

a = np.array([1, 3, 2, 4, 5])
max_value = np.max(a)
max_index = np.argmax(a)
print("最大值为：", max_value)
print("最大值下标为：", max_index)

输出结果为：

最大值为： 5
最大值下标为： 4

对于一个二维数组b，可以使用flatten函数将其转化为一维数组，然后再使用argmax函数获取最大值及其下标。例如：

b = np.array([[1, 2, 3], [4, 5, 6]])
max_value = np.max(b)
max_index = np.argmax(b.flatten())
print("最大值为：", max_value)
print("最大值下标为：", max_index)

输出结果为：

最大值为： 6
最大值下标为： 5