首页 文章

如何使用symPy和numPy替换矩阵的符号

提问于
浏览
6

我试图用我的等式中的两个符号代替它们中的每个矩阵形式 .

我创建了一个换向器函数,它形成了我的表达式:

t, vS, = sy.symbols('t, vS', commutative = False)
hS = t + vS

eta = myComm(t,hS)

dHs = myComm(eta,hS) 
print dHs.expand()

产生我想要的正确表达:

2*t*vS*t + t*vS**2 - t**2*vS - 2*vS*t*vS - vS*t**2 + vS**2*t

所以现在我希望用符号替换符号t和vS,但是当使用subs我得到一个错误时,“unhashable type:'list'”我假设它与我的矩阵初始化或它们应该如何被正确替换有关因为我是numPy和symPy的新手 .

其余代码:

tRel = ([e0, 0],[0,e1])
vtmp = ([v0, v1],[v2,v3])

dHs = dHs.subs(t, tRel)
dHs = dHs.subs(vS, vtmp)
print dHs

1 回答

  • 4

    也许使用lambdify

    import sympy as sy
    import numpy as np    
    from sympy.abc import x, y
    
    z = ((x+y)**2).expand()
    print(z)
    # x**2 + 2*x*y + y**2
    X = np.arange(6).reshape(2,3)
    Y = np.arange(1,7).reshape(2,3)    
    
    f = sy.lambdify((x, y), z, 'numpy')
    print(f(X, Y))
    # [[  1   9  25]
    #  [ 49  81 121]]
    
    assert np.allclose(f(X, Y), (X**2 + 2*X*Y + Y**2))
    

相关问题