cupyx.scipy.signal.argrelextrema#

cupyx.scipy.signal.argrelextrema(data, comparator, axis=0, order=1, mode='clip')[source]#

计算 data 的相对极值。

参数:

  • data (ndarray) – 要查找相对极值的数组。

  • comparator (callable) – 用于比较两个数据点的函数。应接受两个数组作为参数。

  • axis (int, optional) – 沿其从 data 中选择的轴。默认值为 0。

  • order (int, optional) – 用于比较两侧的点数，以便将 comparator(n, n+x) 视为 True。

  • mode (str, optional) – 如何处理向量的边界。可用选项为 'wrap'（环绕）或 'clip'（将溢出视为与最后一个（或第一个）元素相同）。默认值为 'clip'。参见 cupy.take。

返回:

extrema – 整数数组中的极值索引。 extrema[k] 是 data 沿轴 k 的索引数组。注意，即使 data 是一维的，返回值也是一个元组。

返回类型:

tuple，包含 ndarrays

另请参阅

argrelmin, argrelmax

示例

>>> from cupyx.scipy.signal import argrelextrema
>>> import cupy
>>> x = cupy.array([2, 1, 2, 3, 2, 0, 1, 0])
>>> argrelextrema(x, cupy.greater)
(array([3, 6]),)
>>> y = cupy.array([[1, 2, 1, 2],
...               [2, 2, 0, 0],
...               [5, 3, 4, 4]])
...
>>> argrelextrema(y, cupy.less, axis=1)
(array([0, 2]), array([2, 1]))