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

相关问题

numpy求最大值及其下标

在NumPy中，你可以使用`np.max()`函数来找到数组中的最大值，而`np.argmax()`函数则可以返回这个最大值的索引位置。这两个函数都适用于一维或二维数组。

`np.max(arr)`会返回数组`arr`中的最大元素，如果数组有多维，可以指定axis参数来选择沿着哪个轴查找最大值（默认为0，即按行查找）。

例如：

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
max_value = np.max(arr)  # 返回整个数组的最大值
max_index = np.argmax(arr)  # 返回整个数组最大值的索引（对于一维数组）

max_value_axis_0 = np.max(arr, axis=0)  # 沿着列查找每列的最大值
max_index_axis_0 = np.argmax(arr, axis=0)

如果你需要同时获取最大值及其下标，可以这样做：

value_and_index = np.unravel_index(np.argmax(arr), arr.shape)
max_value = value_and_index
max_index = value_and_index

这将返回一个元组，包含最大值所在的行和列的索引。

如何获取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表示沿行方向寻找最大值。