python中的np.argmax() 返回最大值索引号

下面是关于“Python中的np.argmax()返回最大值索引号”的完整攻略，包含了两个示例。

np.argmax()函数

在Python中，可以使用np.argmax()函数返回数组中最大值的索引号。下面是一个示例，演示何使用np.argmax()函数。

import numpy as np

# 创建一维数组
a = np.array([1, 2, 3, 4, 5])

# 返回最大值的索引号
max_index = np.argmax(a)

# 输出结果
print(max_index)

在上面的示例中，我们首先创建了一个一维数组。然后，使用np.argmax()函数返回数组中最大值的索引号。输出结果为：

4

需要注意的是，np.argmax()函数返回的是最大值的索引号，而不是最大值本身。

示例2

下面是另一个示例，演示如何使用np.argmax()函数返回二维数组中每行最大值的引号。

import numpy as np

# 创建一个二维数组
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 返回每行最大值的索引号
max_index = np.argmax(a, axis=1)

# 输出
print(max_index)

在上面的示例中，我们首先创建了一个二维数组a。然后，使用np.argmax()函数返回每行最大值的索引号。需要注意的是，我们使用了axis参数来指定沿着行的方向进行计算。输出结果为：

[2 2 2]

注意的是，np.argmax()函数返回的是最大值的索引号，而不是最大值本身。

总结

本文介绍了如何在Python中使用np.argmax()函数返回数组中最大值的索引号。使用np.argmax()函数可以方便地实现该功能。在使用np.argmax()函数时，需要注意返回结果是最大值的索引号，而不是最大值本身。同时，可以使用axis参数来指定计算方向。