Python numpy.argmax()

numpy.argmax(a, axis=None, out=None)函数返回沿给定轴最大值的索引。

• a: 输入数组
• axis: 需要计算的轴，默认为 None。如果为 None，则返回整个数组的最大值索引。否则，将会沿着轴计算。
• out: 结果数组，这个数组和返回的 argmax 有相同的维度。

示例

import numpy as np

a = np.array([[1,2,3],[4,5,6],[7,8,9]])
print('原数组：')
print(a)

print('\n在整个数组中的最大索引：')
print(np.argmax(a))
 
print('\n沿着axis=0轴的最大索引：')
print(np.argmax(a, axis= 0))
 
print('\n沿着axis=1轴的最大索引：')
print(np.argmax(a, axis= 1))

输出为：

原数组：
[[1 2 3]
 [4 5 6]
 [7 8 9]]

在整个数组中的最大索引：
8

沿着axis=0轴的最大索引：
[2 2 2]

沿着axis=1轴的最大索引：
[2 2 2]

在整个数组中的最大索引是 8，它是这个数组中的第 8 个元素，也就是 9。在沿着轴 0 或轴 1 的计算中，返回的都是沿着轴计算后的最大值索引。

注意事项

• 输入数组的类型应该是数值型，否则可能会出现错误。
• 如果有多个最大值，则只返回第一个最大值的索引。