@@ -1471,8 +1471,7 @@ def _make_pts(self, t, f1, f2, idx0, idx1, step, interpolate):
1471
1471
f2_slice = f2 [idx0 :idx1 ]
1472
1472
if step is not None :
1473
1473
step_func = cbook .STEP_LOOKUP_MAP ["steps-" + step ]
1474
- t_slice , f1_slice , f2_slice = \
1475
- step_func (t_slice , f1_slice , f2_slice )
1474
+ t_slice , f1_slice , f2_slice = step_func (t_slice , f1_slice , f2_slice )
1476
1475
1477
1476
size = len (t_slice )
1478
1477
pts = np .zeros ((2 * size + 2 , 2 ))
@@ -1509,15 +1508,12 @@ def _get_interp_point(t, f1, f2, idx):
1509
1508
return t [im1 ], f1 [im1 ]
1510
1509
elif np .ma .is_masked (diff_values [0 ]):
1511
1510
return t [idx ], f1 [idx ]
1512
- else :
1513
- diff_order = diff_values .argsort ()
1514
- diff_root_ind = np .interp (
1515
- 0 , diff_values [diff_order ], t_values [diff_order ])
1516
- ind_order = t_values .argsort ()
1517
- diff_root_dep = np .interp (
1518
- diff_root_ind ,
1519
- t_values [ind_order ], f1_values [ind_order ])
1520
- return diff_root_ind , diff_root_dep
1511
+
1512
+ diff_order = diff_values .argsort ()
1513
+ diff_root_t = np .interp (0 , diff_values [diff_order ], t_values [diff_order ])
1514
+ t_order = t_values .argsort ()
1515
+ diff_root_f = np .interp (diff_root_t , t_values [t_order ], f1_values [t_order ])
1516
+ return diff_root_t , diff_root_f
1521
1517
1522
1518
def _normalise_pts (self , pts ):
1523
1519
"""
0 commit comments