示例#1
0
文件: base.py 项目: sjhernes/chaospy
def swapaxes(P, ax1, ax2):
    A = P.A.copy()
    if P.dtype==f.frac:
        for key in P.keys:
            A[key] = f.swapaxes(A[key], ax1, ax2)
    else:
        for key in P.keys:
            A[key] = np.swapaxes(A[key], ax1, ax2)
    return Poly(A, P.dim, None, P.dtype)
示例#2
0
def swapaxes(P, ax1, ax2):
    A = P.A.copy()
    if P.dtype == f.frac:
        for key in P.keys:
            A[key] = f.swapaxes(A[key], ax1, ax2)
    else:
        for key in P.keys:
            A[key] = np.swapaxes(A[key], ax1, ax2)
    return Poly(A, P.dim, None, P.dtype)
示例#3
0
def swapaxes(A, ax1, ax2):
    if isinstance(A, (int, long, float, np.ndarray)):
        return np.swapaxes(A, ax1, ax2)

    elif isinstance(A, f.frac):
        return f.swapaxes(A, ax1, ax2)

    elif isinstance(A, p.Poly):
        return p.swapaxes(A, ax1, ax2)

    raise NotImplementedError
示例#4
0
def swapaxes(A, ax1, ax2):
    if isinstance(A, (int, long, float, np.ndarray)):
        return np.swapaxes(A, ax1, ax2)

    elif isinstance(A, f.frac):
        return f.swapaxes(A, ax1, ax2)

    elif isinstance(A, p.Poly):
        return p.swapaxes(A, ax1, ax2)

    raise NotImplementedError