Python numpy matrix.argmin()

简介

matrix.argmin() 是 numpy 矩阵中的方法，用于返回矩阵中最小元素的索引（位置）。如果矩阵中有多个最小元素，matrix.argmin() 将返回第一个最小元素的索引。

用法

matrix.argmin() 的用法非常简单，只需要将矩阵对象调用该方法即可：

import numpy as np

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

index = matrix.argmin()

print(index)

输出：

0

上面的代码中，我们定义了一个 3x3 的矩阵，并将其存储在 matrix 变量中。然后，我们调用了 matrix.argmin() 方法，并将其返回值存储在 index 变量中。最后，我们打印了 index 的值，即最小元素的索引。

注意，matrix.argmin() 方法返回的是一维数组中的索引，因此，如果需要获得矩阵中最小元素的行列位置，可以使用 numpy.unravel_index() 方法进行转换：

row, col = np.unravel_index(index, matrix.shape)
print(row, col)

输出：

0 0

上面的代码中，我们使用 numpy.unravel_index() 方法将一维索引转换为二维坐标，然后打印了最小元素的行列位置。

注意事项

• matrix.argmin() 方法只能用于 numpy 矩阵，不能用于 ndarray 或其他类型的数组。
• 如果矩阵中有多个最小元素，matrix.argmin() 方法将返回第一个最小元素的索引。
• 如果矩阵中的元素类型不是数字类型，matrix.argmin() 方法将会报错。
• 如果矩阵中没有元素，matrix.argmin() 方法将会报错。