Skip to content
Snippets Groups Projects

Base knife-edge scan analysis implementation

Merged Cammille Carinan requested to merge knife-edge-base into master
1 unresolved thread
1 file
+ 3
7
Compare changes
  • Side-by-side
  • Inline
@@ -49,17 +49,13 @@ def prepare_arrays(positions: np.ndarray, intensities: np.ndarray,
@@ -49,17 +49,13 @@ def prepare_arrays(positions: np.ndarray, intensities: np.ndarray,
assert positions.shape == intensities.shape
assert positions.shape == intensities.shape
# Clean both arrays by only getting finite values
# Clean both arrays by only getting finite values
positions, intensities = finite_array(positions, wrt=intensities)
finite_idx = np.isfinite(positions) & np.isfinite(intensities)
intensities, positions = finite_array(intensities, wrt=positions)
positions = positions[finite_idx]
 
intensities = intensities[finite_idx]
return positions, intensities
return positions, intensities
def finite_array(array, *, wrt):
index = np.isfinite(wrt)
return array[index], wrt[index]
def range_mask(array, minimum=None, maximum=None):
def range_mask(array, minimum=None, maximum=None):
default = np.ones(array.shape, dtype=np.bool)
default = np.ones(array.shape, dtype=np.bool)
min_slice, max_slice = default, default
min_slice, max_slice = default, default
Loading