diff --git a/ensembleperturbation/plotting/geometry.py b/ensembleperturbation/plotting/geometry.py index d77084a5..c65ddd1e 100644 --- a/ensembleperturbation/plotting/geometry.py +++ b/ensembleperturbation/plotting/geometry.py @@ -126,10 +126,13 @@ 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()