From 16fb3af354243549f5f4b3096daf736056e926d4 Mon Sep 17 00:00:00 2001 From: Jakub Bartczuk Date: Thu, 31 Oct 2019 18:05:58 +0100 Subject: [PATCH] added min and max --- bootstrapped/stats_functions.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/bootstrapped/stats_functions.py b/bootstrapped/stats_functions.py index a490351..dfeb631 100644 --- a/bootstrapped/stats_functions.py +++ b/bootstrapped/stats_functions.py @@ -38,6 +38,7 @@ def median(values, axis=1): else: return _np.median(_np.asmatrix(values), axis=axis).A1 + def std(values, axis=1): '''Returns the std of each row of a matrix''' if isinstance(values, _sparse.csr_matrix): @@ -45,3 +46,21 @@ def std(values, axis=1): return ret.A1 else: return _np.std(_np.asmatrix(values), axis=axis).A1 + + +def min(values, axis=1): + '''Returns the min of each row of a matrix''' + if isinstance(values, _sparse.csr_matrix): + ret = values.min(axis=axis) + return ret.A1 + else: + return _np.min(_np.asmatrix(values), axis=axis).A1 + + +def max(values, axis=1): + '''Returns the max of each row of a matrix''' + if isinstance(values, _sparse.csr_matrix): + ret = values.max(axis=axis) + return ret.A1 + else: + return _np.max(_np.asmatrix(values), axis=axis).A1