add set_data_3d method to Line3D

Raab70 · Raab70 · commit a92dff47b051 · 2018-06-11T10:37:29.000-05:00
diff --git a/lib/mpl_toolkits/mplot3d/art3d.py b/lib/mpl_toolkits/mplot3d/art3d.py
@@ -151,6 +151,21 @@ def set_3d_properties(self, zs=0, zdir='z'):
         self._verts3d = juggle_axes(xs, ys, zs, zdir)
         self.stale = True
 
+    def set_data_3d(self, *args):
+        """
+        Set the x, y and z data
+
+        ACCEPTS: 3D array (rows are x, y, z) or three 1D arrays
+        """
+        if len(args) == 1:
+            x, y, z = args[0]
+        else:
+            x, y, z = args
+
+        self.set_xdata(x)
+        self.set_ydata(y)
+        self.set_3d_properties(zs=z)
+
     @artist.allow_rasterization
     def draw(self, renderer):
         xs3d, ys3d, zs3d = self._verts3d
