基於numpy用滑窗方差的辦法確定曲線拐點

基於numpy用滑窗方差的辦法確定曲線拐點

需要確定一條抖動的數據的拐點

 

 

#!usr/bin/env python
#-*- coding:utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np
from scipy.signal import savgol_filter


def val_eb(x,y,gap):
# 轉折點前后數據的離散程度相差很大，用這個離散差異擬合一條曲線，可能會有奇效。

# 對每一個數據點（離散數據點擬合的曲線），向前向后分別取一段等長區間，分別求方差，用方差表示離散程度。用向前區間的方差/向后區間的方差，表示離散差異。

# 每個數據點求得的離散差異形成的曲線如下，極值點x=790，正是轉折點。
    col=[]
    y_col = []
    for i in range(gap, len(y) - gap):
        col.append(np.var(y[i - gap:i]) / np.var(y[i:i + gap]))
        y_col.append(i)
    return col,y_col
    pass


def pl():
    n = 1000
    x = np.linspace(1,10,n)

    # 繪制曲線
    noise = np.random.normal(0, 0.008, 1000)
    # print(noise)
    y =np.tanh(x)+noise

    plt.plot(x,y,color="red",linewidth = '0.1')

    # x_smooth = savgol_filter(y, 101, 3)
    y = savgol_filter(y, 201, 3)
    # 方差求拐點
    v,y_v = val_eb(x,y,100)
    i = y_v[v.index(max(v))]
    print("the ebow is ",v,y_v[v.index(max(v))])
    print(x[i],y[i],x[i+1],y[i+1],x[i-1],y[i-1])
    # plt.plot(y_v,v,color="blue")
    plt.plot(x,y,color="green",linewidth = '0.1')
    plt.show()

pl()

能夠識別到x=2.79為拐點，y=0.99