Source code for pebm.ebm.Statistics


import numpy as np


[docs]def statistics(data_dict): stat_dict = {} for i in data_dict: data = data_dict[i] data = data[~np.isnan(data)] if len(data)>0: mean_ = np.mean(data) median_ = np.median(data) min_ = np.amin(data) max_ = np.amax(data) q75, q25 = np.percentile(data, [75, 25]) iqr_ = q75 - q25 std_ = np.std(data) stat_dict[i] = {'mean': mean_, 'median': median_, 'min': min_, 'max': max_, 'iqr': iqr_, 'std': std_} else: stat_dict[i] = {'mean': np.nan, 'median': np.nan, 'min': np.nan, 'max': np.nan, 'iqr': np.nan, 'std': np.nan} return stat_dict