python求极值（极大值、极小值）

使用scipy.signal就可以求得极值点，虽然有时候不太对，但多数情况是对的，核心代码如下：import scipy.signal as sgsg.argrelmax(values) # 极大值的下标sg.argrelmin(values) # 极小值的下标示例import numpy as npimport scipy.signal as sgdef get_maxima(values:

呆萌的代Ma

15224人浏览 · 2021-12-03 17:07:05

呆萌的代Ma · 2021-12-03 17:07:05 发布

使用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))