bucky.util.rolling_mean

Provides generic rolling Pythagorean means over cupy/numpy ndarrays.

Module Contents

Functions

_rolling_arithmetic_mean(arr, window_size=7, axis=0, weights=None)

Compute a rolling arithmetic mean.

_rolling_geometric_mean(arr, window_size, axis=0, weights=None)

Compute a rolling geometric mean.

_rolling_harmonic_mean(arr, window_size, axis=0, weights=None)

Compute a rolling harmonic mean.

rolling_mean(arr, window_size=7, axis=0, weights=None, mean_type='arithmetic')

Calculate a rolling mean over a numpy/cupy ndarray.

rolling_window(a, window_size, center=True, axis=0, pad=True, pad_mode='reflect', reflect_type='odd', freq=1)

Use stride_tricks to add an extra dim on the end of an ndarray for each elements window.

bucky.util.rolling_mean._rolling_arithmetic_mean(arr, window_size=7, axis=0, weights=None)[source]

Compute a rolling arithmetic mean.

bucky.util.rolling_mean._rolling_geometric_mean(arr, window_size, axis=0, weights=None)[source]

Compute a rolling geometric mean.

bucky.util.rolling_mean._rolling_harmonic_mean(arr, window_size, axis=0, weights=None)[source]

Compute a rolling harmonic mean.

bucky.util.rolling_mean.rolling_mean(arr, window_size=7, axis=0, weights=None, mean_type='arithmetic')[source]

Calculate a rolling mean over a numpy/cupy ndarray.

bucky.util.rolling_mean.rolling_window(a, window_size, center=True, axis=0, pad=True, pad_mode='reflect', reflect_type='odd', freq=1)[source]

Use stride_tricks to add an extra dim on the end of an ndarray for each elements window.