numpy获取最大索引

在NumPy中，可以使用numpy.argmax()来获取数组中最大值的索引。该函数返回数组中沿给定轴的最大值的索引。

import numpy as np

# 定义一个一维数组
arr1 = np.array([5, 2, 8, 7, 1])
# 获取arr1中最大值的索引
max_index = np.argmax(arr1)
# 打印结果
print("arr1中最大值的索引为：", max_index)
# 输出：arr1中最大值的索引为： 2

# 定义二维数组
arr2 = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 获取每行最大值的索引
max_index_row = np.argmax(arr2, axis=1)
# 获取每列最大值的索引
max_index_col = np.argmax(arr2, axis=0)
# 打印结果
print("每行最大值的索引为：", max_index_row)
print("每列最大值的索引为：", max_index_col)
# 输出：每行最大值的索引为： [2 2 2]
#      每列最大值的索引为： [2 2 2]

在上面的例子中，我们使用numpy.argmax()函数来获取数组arr1中最大值的索引，结果为2。我们还可以在二维数组（arr2）中使用该函数来获取每行和每列的最大值的索引。

🌟注意事项：当数组中有多个最大值时，numpy.argmax()只返回第一次出现最大值的索引。