binning with NaN values in array

proxtoolbox/Problems/OrbitalTomog/orbitaltomog_data_processor.py

@@ -120,6 +120,12 @@ def bin_2d_array(arr, new_shape):
     """
     shape = (new_shape[0], arr.shape[0] // new_shape[0],
              new_shape[1], arr.shape[1] // new_shape[1])
+    if np.any(np.isnan(arr)):
+        binfactor = 1
+        for i, s in enumerate(arr.shape):
+            binfactor *= new_shape[i] / s
+        return np.nanmean(arr.reshape(shape), axis=(3, 1)) * binfactor
+    else:
         return arr.reshape(shape).sum(-1).sum(1)