lines2gfx: speed-up matching colors to line segments

schlegelp · May 8, 2024 · d3272f9 · d3272f9
1 parent f3220ac
commit d3272f9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/octarine/visuals.py b/octarine/visuals.py
@@ -441,8 +441,8 @@ def lines2gfx(lines, color, linewidth=1, linewidth_space="screen", dash_pattern=
             # See if we can rescue this
             if len(color) == n_points:
                 breaks = np.where(np.isnan(lines[:, 0]))[0]
-                for b in breaks:
-                    color = np.insert(color, b, np.nan, axis=0)
+                offset = np.arange(len(breaks))
+                color = np.insert(color, breaks-offset, np.nan, axis=0)
             else:
                 raise ValueError(f"Got {len(color)} colors for {n_points} points.")
         color = color.astype(np.float32, copy=False)