# Annotate wave numbers (first 5 waves if exist) waves = result['waves'] for i, wave in enumerate(waves[:5]): mid_idx = (wave['start_idx'] + wave['end_idx']) // 2 mid_price = (wave['start_price'] + wave['end_price']) / 2 plt.text(mid_idx, mid_price, str(i+1), fontsize=12, fontweight='bold', bbox=dict(facecolor='yellow', alpha=0.7))
w1, w2, w3, w4, w5 = waves[:5]
return { 'pattern': pattern_type, 'waves': waves, 'valid': impulse_ok or corrective_ok, 'fibonacci_levels': fibs, 'swing_points': swings_df } Example usage & visualization ------------------------------- if name == " main ": import matplotlib.pyplot as plt elliott wave python code
# Mark swing points swings = result['swing_points'] plt.scatter(swings['index'], swings['price'], c='red' if swings['type'].iloc[0]=='high' else 'green', label='Swing points') # Annotate wave numbers (first 5 waves if
swings = sorted(swings, key=lambda x: x['index']) return pd.DataFrame(swings) 'valid': impulse_ok or corrective_ok