如何使用symPy和numPy将符号替换为矩阵
问题内容:
我正在尝试将方程式中的两个符号替换为每个符号的矩阵形式。
我创建了一个换向器函数,该函数形成了我的表达式:
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
问题答案:
也许使用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))