numpy.ndarra取最大值索引

可以使用numpy的argmax函数来取得ndarray中最大值的索引。例如，下面的代码可以取得数组a中最大值的索引：

import numpy as np

a = np.array([1, 3, 2, 4, 5])
max_index = np.argmax(a)

print(max_index)  # 输出 4

如果想在多维数组中取得最大值的索引，可以指定axis参数来沿着指定的维度进行计算。例如，下面的代码可以取得数组b中每一行的最大值的索引：

b = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
max_indices = np.argmax(b, axis=1)

print(max_indices)  # 输出 [2 2 2]