python 平均周期 - time_series转折点的平均持续时间t-p，p-t

Posted 2021-05-10

def avg_periods(peaks, troughs):

    ''' Routine to calculate average duration p-t, t-p
    '''
    
    p_to_t = []
    t_to_p = []
    
    if min(peaks + troughs) in peaks:
         for i in range(len(peaks)):
            try:
                p_to_t.append(troughs[i] - peaks[i])
            except:
                pass
            try:           
                t_to_p.append(peaks[i+1] - troughs[i])
            except:
                pass
    else:
         for i in range(len(troughs)):
            try:
                t_to_p.append(peaks[i] - troughs[i])
            except:
                pass
            try:           
                p_to_t.append(troughs[i+1] - peaks[i])
            except:
                pass

    return np.average(p_to_t), np.average(t_to_p), p_to_t, t_to_p