Ho i dati di una serie. Generazione di dati
date_rng = pd.date_range('2019-01-01', freq='s', periods=400)
df = pd.DataFrame(np.random.lognormal(.005, .5,size=(len(date_rng), 3)),
columns=['data1', 'data2', 'data3'],
index= date_rng)
s = df['data1']
Voglio creare una linea a zig-zag che si collega tra i massimi locali e i minimi locali, che soddisfi la condizione che sull'asse y |highest - lowest value|
di ciascuna linea a zig-zag debba superare una percentuale (diciamo il 20%) della distanza della precedente linea a zig-zag E un valore prestabilito k (diciamo 1.2)
Posso trovare gli extrema locali usando questo codice:
# Find peaks(max).
peak_indexes = signal.argrelextrema(s.values, np.greater)
peak_indexes = peak_indexes[0]
# Find valleys(min).
valley_indexes = signal.argrelextrema(s.values, np.less)
valley_indexes = valley_indexes[0]
# Merge peaks and valleys data points using pandas.
df_peaks = pd.DataFrame({'date': s.index[peak_indexes], 'zigzag_y': s[peak_indexes]})
df_valleys = pd.DataFrame({'date': s.index[valley_indexes], 'zigzag_y': s[valley_indexes]})
df_peaks_valleys = pd.concat([df_peaks, df_valleys], axis=0, ignore_index=True, sort=True)
# Sort peak and valley datapoints by date.
df_peaks_valleys = df_peaks_valleys.sort_values(by=['date'])
ma non so come applicare la condizione di soglia ad esso. Per favore, mi consigli su come applicare tale condizione.
Poiché i dati potrebbero contenere milioni di timestamp, è altamente raccomandato un calcolo efficiente
Per una descrizione più chiara:
Esempio di output, dai miei dati:
# Instantiate axes.
(fig, ax) = plt.subplots()
# Plot zigzag trendline.
ax.plot(df_peaks_valleys['date'].values, df_peaks_valleys['zigzag_y'].values,
color='red', label="Zigzag")
# Plot original line.
ax.plot(s.index, s, linestyle='dashed', color='black', label="Org. line", linewidth=1)
# Format time.
ax.xaxis_date()
ax.xaxis.set_major_formatter(mdates.DateFormatter("%Y-%m-%d"))
plt.gcf().autofmt_xdate() # Beautify the x-labels
plt.autoscale(tight=True)
plt.legend(loc='best')
plt.grid(True, linestyle='dashed')
L'output desiderato (qualcosa di simile a questo, lo zigzag collega solo i segmenti significativi)