SymPy向量减法结果不符合预期，请求排查代码问题

我编写了一段使用SymPy进行向量计算的代码，但运行结果与数学预期不符。代码如下：

import sympy as sp
from sympy.vector import CoordSys3D

N = CoordSys3D('N')
t = sp.symbols('t')

r = sp.symbols('r')
w = sp.symbols('w')
o = sp.symbols('o')


x = sp.Function('x')(t)
y = sp.Function('y')(t)
z = sp.Function('z')(t)

p = x*N.i + y*N.j + z*N.k

q = r*(p.dot(N.i)*sp.cos(w*t)+p.dot(N.j)*sp.sin(w*t))+o*p.dot(N.k)
print(p-q)

运行结果为：

(x(t))*N.i + (y(t))*N.j + (z(t))*N.k

但从数学角度，我预期的结果应该是：

x(t)-r*x(t)*sp.cos(w*t) + y(t)-r*y(t)*sp.sin(w*t) + z(t)-o*z(t)

请问我哪里出错了？是否忽略了某些细节？

你的核心问题是**q的定义不符合向量运算规则**：

• 当前q是一个标量：p.dot(N.i)等是标量，所有标量运算的结果还是标量。而p是向量，SymPy中向量与标量直接做减法是不合法的，它会直接返回原向量，这就是你看到结果未变化的原因。
• 你需要将q定义为向量，每个分量对应预期的计算逻辑，绑定到对应的坐标系基向量上。

修改后的代码：

import sympy as sp
from sympy.vector import CoordSys3D

N = CoordSys3D('N')
t = sp.symbols('t')

r = sp.symbols('r')
w = sp.symbols('w')
o = sp.symbols('o')


x = sp.Function('x')(t)
y = sp.Function('y')(t)
z = sp.Function('z')(t)

p = x*N.i + y*N.j + z*N.k

# 修正q的定义，将每个分量对应到向量基上
q = r*x*sp.cos(w*t)*N.i + r*y*sp.sin(w*t)*N.j + o*z*N.k
print(p - q)

运行后输出结果与你的数学预期一致：

(x(t) - r*x(t)*cos(t*w))*N.i + (y(t) - r*y(t)*sin(t*w))*N.j + (z(t) - o*z(t))*N.k

补充：p.dot(N.i)等价于x(t)，p.dot(N.j)等价于y(t)，p.dot(N.k)等价于z(t)，直接使用变量x、y、z可以简化代码，无需调用dot方法。

内容的提问来源于stack exchange，提问作者Mo711