python argmax

argmax函数是Python中的一个常用函数，用于返回数组中最大值的索引。在NumPy库中，可以使用argmax函数来找到数组中最大的索引。

以下是一个示例：

import numpy as np

arr np.array([1, 5, 3, 8, 2])
max_index = np.argmax(arr)

print("最大值的索引为:", max_index)

输出结果为:

最大值的索引为: 3

在这个例子中，数组arr中的最大值是8，它的索引是3。argmax函数返回了最大值8所在的索引3。