Step Function with Points

import matplotlib.pyplot as plt
import numpy as np

# === STYLE CONFIG ===
COLORS = {
    'step': '#10B981',       # Emerald
    'markers': '#F59E0B',    # Amber
    'background': '#FAFBFC',
    'text_muted': '#64748B',
    'grid': '#E2E8F0',
}

# === DATA ===
x = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
y = np.array([1, 1.5, 1.5, 3, 3, 2.5, 4, 4, 5.5, 5.5, 6])

# === FIGURE ===
fig, ax = plt.subplots(figsize=(10, 6), dpi=100)
ax.set_facecolor(COLORS['background'])
fig.patch.set_facecolor(COLORS['background'])

# === PLOT ===
ax.step(x, y, where='post', 
        color=COLORS['step'], linewidth=2.5,
        label='Level', zorder=2)

ax.scatter(x, y, 
           color=COLORS['markers'], s=70,
           edgecolors='white', linewidths=2,
           label='Change points', zorder=3)

# === AXES ===
ax.set_xlim(-0.5, 10.5)
ax.set_ylim(0, 7.5)
ax.set_xlabel('Stage', fontsize=10, color=COLORS['text_muted'], labelpad=10)
ax.set_ylabel('Level', fontsize=10, color=COLORS['text_muted'], labelpad=10)

# === STYLING ===
for spine in ['top', 'right']:
    ax.spines[spine].set_visible(False)
for spine in ['bottom', 'left']:
    ax.spines[spine].set_color(COLORS['grid'])

ax.yaxis.grid(True, color=COLORS['grid'], linewidth=0.5, alpha=0.7)
ax.xaxis.grid(False)
ax.set_axisbelow(True)
ax.tick_params(axis='both', colors=COLORS['text_muted'], labelsize=9, length=0, pad=8)

ax.legend(loc='upper center', bbox_to_anchor=(0.5, -0.12),
          ncol=2, frameon=False, fontsize=9, labelcolor=COLORS['text_muted'])

plt.tight_layout()
plt.show()