如何获取np数组中最大值下标

可以使用numpy库中的argmax函数来获取np数组中最大值的下标。具体代码如下所示：

import numpy as np

arr = np.array([1, 5, 3, 9, 2])
max_index = np.argmax(arr)
print(max_index)  # 输出4

如果是二维数组，可以加上axis参数来指定沿哪个维度寻找最大值，代码如下所示：

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
max_index = np.argmax(arr, axis=1)
print(max_index)  # 输出[2 2 2]

其中，axis=0表示沿列方向寻找最大值，axis=1表示沿行方向寻找最大值。

相关问题

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

python获取数组中最大的三个值的下标

可以使用numpy库中的argsort函数来实现，代码如下：

import numpy as np

arr = np.array([1, 3, 2, 5, 4])
indices = arr.argsort()[-3:][::-1]
print(indices)

输出结果为：

[3 4 1]