argmax 用于 2d 数组

在 Numpy 库中，argmax 函数用于查找数组中最大元素的索引或者沿着轴返回最大值的索引。当应用于 2D 数组时，argmax 函数的使用方法与 1D 数组的使用方法类似。

函数签名

numpy.argmax(arr, axis=None)

参数说明

• arr: 数组
• axis: 默认为 None。如果指定了轴，函数将沿着轴查找最大值。

返回值说明

• 如果 axis=None，则返回所有元素中最大值的索引。
• 如果 axis=0，则返回沿着第一个轴的最大值的索引。
• 如果 axis=1，则返回沿着第二个轴的最大值的索引。

例子

import numpy as np

a = np.array([[1, 2, 3],
              [4, 5, 6],
              [7, 8, 9]])

print(np.argmax(a)) # 输出: 8，因为9是数组中最大值，它的索引为8（拉成1D数组之后的索引）
print(np.argmax(a, axis=0)) # 输出: [2 2 2]，因为每一列最大值对应的索引分别为2、2、2
print(np.argmax(a, axis=1)) # 输出: [2 2 2]，因为每一行最大值对应的索引分别为2、2、2

注意事项

• 如果最大值在数组中出现多次，则返回第一个匹配项的索引。
• 当 axis=0 时，表示按列进行查找；当 axis=1 时，表示按行进行查找。
• 当 arr 是一个空数组，函数将返回 0。