"每组获取argmin"是一种Numpy方法,用于在多维数组中找到每组的最小值的索引。
具体来说,当我们需要从一个多维数组中找到每组的最小值所在的位置时,可以使用该方法。每组通常是指数组中的一维子数组,也可以理解为行或列。
该方法的语法如下:
numpy.argmin(a, axis=None, out=None)
参数说明:
a
:待查找的数组。axis
:指定在哪个轴上查找,默认为 None,表示在整个数组中查找最小值的索引。若指定某个轴,比如 axis=0,则表示在每一列中查找最小值的索引。out
:指定输出结果的数组,通常不需要使用。该方法返回一个由每组最小值索引组成的数组,其形状与输入数组的形状相匹配,但是最后一个轴被移除。
使用案例:
import numpy as np
arr = np.array([[3, 4, 5], [1, 2, 3], [6, 7, 8]])
argmin_result = np.argmin(arr, axis=1)
print(argmin_result)
输出:
[0 0 0]
解释:对于二维数组 arr
,每一行的最小值分别是 3、1 和 6,它们所在的索引分别是 0、0 和 0。
应用场景:
推荐的腾讯云相关产品和产品介绍链接地址:
领取专属 10元无门槛券
手把手带您无忧上云