argmin

返回沿指定轴的最小值的索引。当最小值多次出现时，只返回对应于第一次出现的索引。

参数

名称	类型	描述	默认值
x	SparseArray	输入数组。填充值必须为 0.0，所有非零值必须小于 0.0。	必需
axis	int	搜索的轴。如果为 None，函数必须返回扁平化数组的最小值的索引。默认值：None。	None
keepdims	bool_	如果为 True，缩减的轴（维度）必须作为单例维度包含在结果中，并且结果必须与输入数组兼容。否则，如果为 False，缩减的轴（维度）不得包含在结果中。默认值：False。	False

返回值

名称	类型	描述
out	ndarray	如果 axis 为 None，则为包含最小值第一次出现索引的零维数组。否则，为包含最小值索引的非零维数组。

源代码位于 sparse/numba_backend/_coo/common.py

def argmin(x, /, *, axis=None, keepdims=False):
    """
    Returns the indices of the minimum values along a specified axis.
    When the minimum value occurs multiple times, only the indices
    corresponding to the first occurrence are returned.

    Parameters
    ----------
    x : SparseArray
        Input array. The fill value must be ``0.0`` and all non-zero values
        must be less than ``0.0``.
    axis : int, optional
        Axis along which to search. If ``None``, the function must return
        the index of the minimum value of the flattened array. Default: ``None``.
    keepdims : bool, optional
        If ``True``, the reduced axes (dimensions) must be included in the result
        as singleton dimensions, and, accordingly, the result must be compatible
        with the input array. Otherwise, if ``False``, the reduced axes (dimensions)
        must not be included in the result. Default: ``False``.

    Returns
    -------
    out : numpy.ndarray
        If ``axis`` is ``None``, a zero-dimensional array containing the index of
        the first occurrence of the minimum value. Otherwise, a non-zero-dimensional
        array containing the indices of the minimum values.
    """
    return _arg_minmax_common(x, axis=axis, keepdims=keepdims, mode="min")