fix quiver3d incorrect arrow colors

AlecVercruysse · AlecVercruysse · commit e1e1a32c0502 · 2024-02-07T12:01:34.000-08:00
diff --git a/lib/mpl_toolkits/mplot3d/axes3d.py b/lib/mpl_toolkits/mplot3d/axes3d.py
@@ -2986,14 +2986,6 @@ def calc_arrows(UVW):
         # Normalize rows of UVW
         norm = np.linalg.norm(UVW, axis=1)
 
-        # If any row of UVW is all zeros, don't make a quiver for it
-        mask = norm > 0
-        XYZ = XYZ[mask]
-        if normalize:
-            UVW = UVW[mask] / norm[mask].reshape((-1, 1))
-        else:
-            UVW = UVW[mask]
-
         if len(XYZ) > 0:
             # compute the shaft lines all at once with an outer product
             shafts = (XYZ - np.multiply.outer(shaft_dt, UVW)).swapaxes(0, 1)
@@ -3006,7 +2998,7 @@ def calc_arrows(UVW):
             # transpose to get a list of lines
             heads = heads.swapaxes(0, 1)
 
-            lines = [*shafts, *heads]
+            lines = [*shafts, *heads[::2], *heads[1::2]]
         else:
             lines = []
 
