fixed sorting error in exact vol 1D

lcgraham · May 9, 2016 · b6efc02 · b6efc02
1 parent 5fca1d3
commit b6efc02
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/bet/calculateP/calculateP.py b/bet/calculateP/calculateP.py
@@ -424,7 +424,9 @@ def exact_volume_1D(samples, input_domain, distribution='uniform', a=None,
         edges = scipy.stats.beta.cdf(edges, a=a, b=b,
                 loc=input_domain[:, 0], scale=domain_width)
     # calculate difference between right and left of each cell and renormalize
-    lam_vol = np.squeeze(edges[1:, :] - edges[:-1, :])
+    sorted_lam_vol = np.squeeze(edges[1:, :] - edges[:-1, :])
+    lam_vol = np.zeros(sorted_lam_vol.shape)
+    lam_vol[sort_ind] = sorted_lam_vol
     if distribution == 'uniform':
         lam_vol = lam_vol/domain_width
     # Set up local arrays for parallelism