當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文

Python cusignal.peak_finding.peak_finding.argrelextrema用法及代碼示例

用法: cusignal.peak_finding.peak_finding.argrelextrema(data, comparator, axis=0, order=1, mode='clip')

計算 data 的相對極值。

參數：

data：ndarray: 在其中查找相對極值的數組。
comparator：可調用的: 用於比較兩個數據點的函數。應該將兩個數組作為參數。
axis：整數，可選: 從 data 中選擇的軸。默認值為 0。
order：整數，可選: 每邊有多少點用於比較認為comparator(n, n+x)為真。

返回：

extrema：ndarray 的元組: 整數數組中最大值的索引。 extrema[k] 是 data 的軸 k 的索引數組。請注意，即使data 是一維的，返回值也是一個元組。

例子：

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

相關用法

注：本文由純淨天空篩選整理自rapids.ai大神的英文原創作品 cusignal.peak_finding.peak_finding.argrelextrema。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。