コード例 #1
0
ファイル: base.py プロジェクト: sjhernes/chaospy
def sum(P, axis=None):

    if not P.A:
        if axis is None:
            return Poly({}, P.dim, (), P.dtype)
        shape = P.shape[:axis]+P.shape[axis+1:]
        return Poly({}, P.dim, shape, P.dtype)

    if isinstance(axis, int):

        l = len(P.shape)
        if axis<0: axis += l

        shape = [0]*(l-1)
        for i in range(l):
            if i<axis:
                shape[i] = P.shape[i]
            elif i>axis:
                shape[i-1] = P.shape[i]
        shape = tuple(shape)
    else:
        shape = ()

    A = P.A
    if P.dtype==f.frac:
        for key in P.keys:
            A[key] = f.sum(A[key], axis)
    else:
        for key in P.keys:
            A[key] = np.sum(A[key], axis)

    return Poly(A, P.dim, shape, P.dtype)
コード例 #2
0
def sum(P, axis=None):

    if not P.A:
        if axis is None:
            return Poly({}, P.dim, (), P.dtype)
        shape = P.shape[:axis] + P.shape[axis + 1:]
        return Poly({}, P.dim, shape, P.dtype)

    if isinstance(axis, int):

        l = len(P.shape)
        if axis < 0: axis += l

        shape = [0] * (l - 1)
        for i in range(l):
            if i < axis:
                shape[i] = P.shape[i]
            elif i > axis:
                shape[i - 1] = P.shape[i]
        shape = tuple(shape)
    else:
        shape = ()

    A = P.A
    if P.dtype == f.frac:
        for key in P.keys:
            A[key] = f.sum(A[key], axis)
    else:
        for key in P.keys:
            A[key] = np.sum(A[key], axis)

    return Poly(A, P.dim, shape, P.dtype)
コード例 #3
0
ファイル: wrapper.py プロジェクト: nixnfg/integrated
def sum(A, axis=None):
    """
Sum the components of a shapeable quantity along a given axis.

Parameters
----------
A : Poly, frac, array_like
    Input data.
axis : int, optional
    Axis over which the sum is taken. By default `axis` is
    None, and all elements are summed.

Returns
-------
Q : Poly, frac, array_like
    Polynomial array with same shape as `P`, with the specified
    axis removed. If `P` is an 0-d array, or `axis` is None, a
    (non-iterable) component is returned.

Examples
--------
>>> P = cp.prange(3)
>>> print P
[1, q0, q0^2]
>>> print cp.sum(P)
q0^2+q0+1
    """
    if isinstance(A, (np.ndarray, float, int, long)):
        return np.sum(A, axis)

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

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

    raise NotImplementedError
コード例 #4
0
ファイル: wrapper.py プロジェクト: apetcho/chaospy
def sum(A, axis=None):
    """
Sum the components of a shapeable quantity along a given axis.

Parameters
----------
A : Poly, frac, array_like
    Input data.
axis : int, optional
    Axis over which the sum is taken. By default `axis` is
    None, and all elements are summed.

Returns
-------
Q : Poly, frac, array_like
    Polynomial array with same shape as `P`, with the specified
    axis removed. If `P` is an 0-d array, or `axis` is None, a
    (non-iterable) component is returned.

Examples
--------
>>> P = cp.prange(3)
>>> print P
[1, q0, q0^2]
>>> print cp.sum(P)
q0^2+q0+1
    """
    if isinstance(A, (np.ndarray, float, int, long)):
        return np.sum(A, axis)

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

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

    raise NotImplementedError