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

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

官方的文档位置：https://pandas.pydata.org/docs/reference/api/pandas.DataFrame.rolling.html

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