

def get_waveform_amplitude(mean_masks, mean_waveforms):
    'Return the amplitude of the waveforms on all channels.'
    assert (mean_waveforms.ndim == 2)
    (n_samples, n_channels) = mean_waveforms.shape
    assert (mean_masks.ndim == 1)
    assert (mean_masks.shape == (n_channels,))
    mean_waveforms = (mean_waveforms * mean_masks)
    assert (mean_waveforms.shape == (n_samples, n_channels))
    (m, M) = (mean_waveforms.min(axis=0), mean_waveforms.max(axis=0))
    return (M - m)

dataset/ETHPy150Open kwikteam/phy/phy/stats/clusters.py get_waveform_amplitude/original