def mean(P, ax=None): A = P.A.copy() if P.dtype==f.frac: for key in P.keys: A[key] = f.mean(A[key], ax) else: for key in P.keys: A[key] = np.mean(A[key], ax) return Poly(A, P.dim, A[key].shape, P.dtype)
def mean(P, ax=None): A = P.A.copy() if P.dtype == f.frac: for key in P.keys: A[key] = f.mean(A[key], ax) else: for key in P.keys: A[key] = np.mean(A[key], ax) return Poly(A, P.dim, A[key].shape, P.dtype)
def mean(A, ax=None): if isinstance(A, (int, long, float, np.ndarray)): return np.mean(A, ax) elif isinstance(A, f.frac): return f.mean(A) elif isinstance(A, p.Poly): return p.mean(A) raise NotImplementedError