adjust colors

cwhanse · Sep 10, 2024 · ba1694f · ba1694f
1 parent 35cbb7c
commit ba1694f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/docs/examples/snow-detection/snow-mode.py b/docs/examples/snow-detection/snow-mode.py
@@ -656,8 +656,9 @@ def wrapper(voltage, current, temp_cell, effective_irradiance,
 handles.append(measured_line)
 handles.append(modeled_line)
 
-for i in range(-1, N-1):
-    my_patch = mpatches.Patch(color=cmap.colors[i], label=f'Mode {i}')
+for i in [-1, 0, 1, 2, 3, 4]: # modes
+    color_idx = i + 1
+    my_patch = mpatches.Patch(color=cmap.colors[color_idx], label=f'Mode {i}')
     handles.append(my_patch)
 
 ax.set_xlabel('Date', fontsize='xx-large')