使用scipy.signal就可以求得极值点,虽然有时候不太对,但多数情况是对的,核心代码如下:

官方的文档位置:https://docs.scipy.org/doc/scipy/reference/signal.html#peak-finding

import scipy.signal as sg

sg.argrelmax(values) # 极大值 的下标
sg.argrelmin(values) # 极小值 的下标

示例

import numpy as np
import scipy.signal as sg


def get_maxima(values: np.ndarray):
    """极大值点"""
    max_index = sg.argrelmax(values)[0]
    return max_index, values[max_index]


def get_minima(values: np.ndarray):
    """极小值点"""
    min_index = sg.argrelmin(values)[0]  # 极小值的下标
    return min_index, values[min_index]  # 返回极小值


if __name__ == '__main__':
    data = np.array([2, 1.5, 1, 1.5, 2, 3, 2, 0, 2, 3, 0])
    print("极值点下标", "极值")
    print(get_minima(data))
    print(get_maxima(data))
Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐