argmax函数python

作者：整整ー世♂陰霾そ发布日期:2026-05-20 浏览:128

import numpy as np

# 示例数组
arr = np.array([1, 3, 2, 5, 4])

# 使用 argmax 函数找到最大值的索引
index_of_max = np.argmax(arr)

print(f"数组: {arr}")
print(f"最大值的索引: {index_of_max}")

解释说明：

np.argmax(arr)：该函数返回数组 arr 中最大值的索引。如果数组中有多个相同的最大值，它会返回第一个出现的最大值的索引。
在上面的例子中，数组 [1, 3, 2, 5, 4] 中的最大值是 5，它的索引是 3，因此 np.argmax(arr) 返回 3。

如果你需要在多维数组中使用 argmax，可以指定 axis 参数来控制沿着哪个轴进行操作。例如：

# 多维数组示例
arr_2d = np.array([[1, 3], [2, 5]])

# 沿着 axis=0（列方向）寻找最大值的索引
index_of_max_axis0 = np.argmax(arr_2d, axis=0)

# 沿着 axis=1（行方向）寻找最大值的索引
index_of_max_axis1 = np.argmax(arr_2d, axis=1)

print(f"二维数组: \n{arr_2d}")
print(f"沿列方向的最大值索引: {index_of_max_axis0}")
print(f"沿行方向的最大值索引: {index_of_max_axis1}")

上一篇：python replace函数

下一篇：python api接口开发