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