📅  最后修改于: 2023-12-03 14:46:38.176000             🧑  作者: Mango
在Python的数据处理和科学计算中,numpy库是非常常见的工具之一。在numpy库中,numpy.argmin函数作为常见的数组操作函数,常常用于查找数组中最小值所在的位置。
在numpy库中,argmin()函数是用于在给定轴上查找最小值所在的位置的函数。
numpy.argmin(arr, axis=None, out=None)
arr
:输入数组axis
:int类型的可选参数,表示查找的轴方向,如果没有该参数,函数会在整个数组中查找最小值的位置。out
:数组类型的可选参数,目标输出数组,函数将返回查找最小值的位置。argmin()函数返回的是指定axis轴上最小值所在的下标。
import numpy as np
a = np.array([[1,2,3], [4,5,6], [7,8,9]])
print(np.argmin(a)) # 输出:0
print(np.argmin(a, axis=0)) # 输出:[0 0 0]
print(np.argmin(a, axis=1)) # 输出:[0 0 0]
上述代码中,我们使用numpy库的argmin()函数来查找多维数组a中最小值所在的位置。
第一次使用argmin()函数时,没有给定指定轴位置的参数,所以默认函数会在整个数组中查找最小值的下标,即坐标(0,0)的位置。
第二次使用argmin()函数给定参数axis=0时,函数将沿着纵轴查找最小值所在的位置,结果是一个长度为3的下标数组[0,0,0],代表在纵轴上,第0列(第一列)上的元素都是最小值。
第三次使用argmin()函数给定参数axis=1时,函数将沿着横轴查找最小值所在的位置,结果是一个长度为3的下标数组[0,0,0],代表在横轴上,第0行(第一行)上的元素都是最小值。
numpy.argmin()函数是numpy库中常用的数组操作函数之一,其能够帮助程序员快速查找数组中最小值所在的位置,方便我们进行下一步的数组处理。在使用时,需要注意函数的语法和参数,特别是axis参数能够控制查找最小值所在的轴位置。