Python中的 numpy.argmin()

numpy.argmin()是numpy库中的一个函数，用于返回数组中最小值的索引值。

语法

numpy.argmin(arr, axis=None, out=None)

• arr：要查找最小值的数组。
• axis：用于排序的轴。默认是仅仅展开数组。
• out：返回的输出数组。

参数

arr

arr参数可以是包含数值的任何可迭代对象，如列表、元组或数组。

import numpy as np

arr = [23, 20, 15, 13, 16, 18]
print(np.argmin(arr))

输出结果为：

3

因此，可以看出在将arr输入到np.argmin()函数中，返回的最小值的索引是3。

axis

当涉及到多维数组时，需要使用axis参数指定轴。如果为None（默认值），则轴在展平的数组中计算。

arr = [[10, 20, 30], [40, 50, 60], [70, 80, 90]]
print(np.argmin(arr, axis=1))

输出结果为：

[0 0 0]

在该语句中，np.argmin()函数沿着每行找到最小值的列索引。

out

指定输出数组后，函数将在该数组中返回结果，而不是在一个新的数组中返回。

arr = [[10, 20, 30], [40, 50, 60], [70, 80, 90]]
out_arr = np.empty(3, dtype=int)
np.argmin(arr, axis=0, out=out_arr)
print(out_arr)

输出结果为：

[0 0 0]

在该语句中，在经过排序的第一维上查找最小值，并返回结果。最小值索引被赋给了预定义的序列out_arr。

返回值

numpy.argmin()函数将返回最小值的整数索引。如果输入不为向量，则返回的是由这些最小值索引组成的新数组，其形状与输入相同。

arr = [[10, 20, 30], [40, 50, 60], [70, 80, 90]]
print(np.argmin(arr))

输出结果为：

0

在该语句中，函数在将arr输入到np.argmin()函数中，返回的数组中最小值的索引是0。