Python | numpy matrix.argmax()

概述

numpy.matrix.argmax() 是 numpy 库中的一个函数，该函数用于返回矩阵中最大值的索引。

语法

numpy.matrix.argmax(axis=None, out=None)

参数

• axis: 需要统计的轴，默认为 None，表示统计所有元素的最大值索引；
• out: 返回的结果存储到指定的输出数组中。

返回值

该函数返回矩阵中最大值的索引。

示例

代码示例

以下示例演示了如何使用 numpy.matrix.argmax() 函数：

import numpy as np

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

# 打印数组
print("Array:\n", a)

# 调用 argmax() 函数
max_index = a.argmax()

# 输出结果
print("Index of max element:", max_index)

输出结果

Array:
 [[1 2 3]
 [4 5 6]
 [7 8 9]]
Index of max element: 8

注意事项

• 在二维矩阵中，使用 numpy.ndarray.argmax() 与 numpy.matrix.argmax() 的结果是一致的；
• 如果矩阵中有重复的最大值，numpy.matrix.argmax() 函数只会返回第一个最大值的索引。