Fix point plot color

noaa-ocs-modeling · Mar 5, 2024 · 7a495a1 · 7a495a1
1 parent 3c485be
commit 7a495a1
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/ensembleperturbation/plotting/geometry.py b/ensembleperturbation/plotting/geometry.py
@@ -126,10 +126,14 @@ def plot_points(
     :param title: whether to add a title to the plot
     """
 
-    if isinstance(points, Iterable):
-        if not isinstance(points, MultiPoint):
-            points = MultiPoint(points)
-        points = numpy.squeeze(numpy.stack((point._get_coords() for point in points), axis=0))
+    if isinstance(points, MultiPoint):
+        points = numpy.squeeze(
+                numpy.stack(
+                    [list(point.coords)[0][:3] for point in points.geoms],
+                    axis=0))
+
+    if not isinstance(points, numpy.ndarray):
+        points = numpy.array(points)
 
     if axis is None:
         axis = pyplot.gca()