Ensure NaN values properly handled in sorted

tomneep · tomneep · commit e66cd0fb803f · 2021-07-14T14:26:18.000+01:00
diff --git a/lib/mpl_toolkits/mplot3d/art3d.py b/lib/mpl_toolkits/mplot3d/art3d.py
@@ -8,6 +8,7 @@
 """
 
 import math
+import sys
 
 import numpy as np
 
@@ -733,7 +734,13 @@ def set_zsort(self, zsort):
             The function applied on the z-coordinates of the vertices in the
             viewer's coordinate system, to determine the z-order.
         """
-        self._zsortfunc = self._zsort_functions[zsort]
+        def nansafe(func):
+            def f(x):
+                value = func(x)
+                return sys.maxsize if np.isnan(value) else value
+            return f
+
+        self._zsortfunc = nansafe(self._zsort_functions[zsort])
         self._sort_zpos = None
         self.stale = True
 
