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.

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.