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

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