python 转折点

Posted 2021-05-10

peaks, troughs = pivots(s, 10)

fig1 = plt.figure(figsize=(8,5))
#plt.plot(s)
s.plot()

for p in peaks:
    plt.plot(s.index[p], s[p], 'ro', markersize=8)
    plt.text(s.index[p], s[p], str(p), fontsize=10)
for t in troughs:
    plt.plot(s.index[t], s[t], 'go', markersize=8)
    plt.text(s.index[t], s[t], str(t), fontsize=10)
    
alt_peaks, alt_troughs = alternate_pivots(s, peaks, troughs)
    
fig2 = plt.figure(figsize=(8,5))
s.plot()

for p in alt_peaks:
    plt.plot(s.index[p], s[p], 'ro', markersize=8)
    plt.text(s.index[p], s[p], str(p), fontsize=10)
for t in alt_troughs:
    plt.plot(s.index[t], s[t], 'go', markersize=8)
    plt.text(s.index[t], s[t], str(t), fontsize=10)