Python cumsum Exemples

Exemple #1

Fichier : base.py Projet : sjhernes/chaospy

def cumsum(P, axis=None):

    A = P.A.copy()
    if P.dtype==f.frac:
        for key in P.keys:
            A[key] = f.cumsum(A[key], axis)
    else:
        for key in P.keys:
            A[key] = np.cumsum(A[key], axis)
    return Poly(A, P.dim, P.shape, P.dtype)

Exemple #2

def cumsum(P, axis=None):

    A = P.A.copy()
    if P.dtype == f.frac:
        for key in P.keys:
            A[key] = f.cumsum(A[key], axis)
    else:
        for key in P.keys:
            A[key] = np.cumsum(A[key], axis)
    return Poly(A, P.dim, P.shape, P.dtype)

Exemple #3

Fichier : wrapper.py Projet : nixnfg/integrated

def cumsum(A, axis=None):

    if isinstance(A, (int, long, float, np.ndarray)):
        return np.cumsum(A, axis)

    elif isinstance(A, f.frac):
        return f.cumsum(A, axis)

    elif isinstance(A, p.Poly):
        return p.cumsum(A, axis)

    raise NotImplementedError

Exemple #4

Fichier : wrapper.py Projet : apetcho/chaospy

def cumsum(A, axis=None):

    if isinstance(A, (int, long, float, np.ndarray)):
        return np.cumsum(A, axis)

    elif isinstance(A, f.frac):
        return f.cumsum(A, axis)

    elif isinstance(A, p.Poly):
        return p.cumsum(A, axis)

    raise NotImplementedError