def mean_var(data): # TODO: assert is a np.array mean = np.nanmean(data, axis=0) var = np.nanvar(data, axis=0) return [mean, var]