fixed bug for guessing datalims

Chilipp · May 2, 2019 · 9be5046 · 9be5046
1 parent 85f6048
commit 9be5046
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/straditize/straditizer.py b/straditize/straditizer.py
@@ -640,7 +640,8 @@ def guess_data_lims(self, fraction=0.7):
             ymax, xmax = right[
                 :, right.shape[1] - 1 - right.max(0)[::-1].argmax()]
         else:
-            xmax = len(cum_y) - 1 - (cum_y[::-1] > max_v).any(axis=0).argmax()
+            xmax = cum_y.shape[1] - 1 - (cum_y[:, ::-1] > max_v).any(
+                axis=0).argmax()
             ymax = len(cum_x) - 1 - (cum_x[::-1] > max_h).any(axis=1).argmax()
 
         left = np.vstack(np.where((cum_xr > max_h) & (cum_yr > max_v)))