Sympy从矢量字段中获取矢量

时间:2018-05-13 18:19:02

标签: python math sympy

我正在使用sympy(非常棒)并且我刚刚创建了一个像这样的矢量字段

> import sympy
> from sympy.vector import CoordSys3D
> from sympy import *
> R = CoordSys3D('R')
> x, y, z, t = symbols('x y z t')
> v = x*R.i + 4*z*R.j + y*R.k
x*R.i + 4*z*R.j + y*R.k
> v.evalf(subs={x:6, y:5, z:2})
6.00000000000000*R.i + 8.00000000000000*R.j + 5.00000000000000*R.k

我需要的是获取表单[6.0,8.0,5.0]的向量或列表,那么有没有办法获得列表形式v.evalf()?我可以在"6.00000000000000*R.i + 8.00000000000000*R.j + 5.00000000000000*R.k"上使用split split或者其他东西,但那看起来很难看,也许有内置的方法

1 个答案:

答案 0 :(得分:1)

In [252]: vector = v.evalf(subs={x:6, y:5, z:2}); vector
Out[252]: 6.00000000000000*R.i + 8.00000000000000*R.j + 5.00000000000000*R.k

In [253]: list(vector.to_matrix(R))
Out[253]: [6.00000000000000, 8.00000000000000, 5.00000000000000]

其他可能性包括

In [256]: vector.as_poly().coeffs()
Out[256]: [6.00000000000000, 8.00000000000000, 5.00000000000000]

In [257]: list(vector.components.values())
Out[257]: [5.00000000000000, 8.00000000000000, 6.00000000000000]

但我认为它们会遇到一个致命的缺陷,当一个或多个组件等于0时会暴露出来。例如,如果z设置为0:

In [258]: vector = v.evalf(subs={x:6, y:5, z:0}); vector
Out[258]: 6.00000000000000*R.i + 5.00000000000000*R.k

然后list(vector.to_matrix(R))仍会返回3个组件:

In [259]: list(vector.to_matrix(R))
Out[259]: [6.00000000000000, 0, 5.00000000000000]

而其他两个表达式省略了零分量:

In [260]: vector.as_poly().coeffs()
Out[260]: [6.00000000000000, 5.00000000000000]

In [261]: list(vector.components.values())
Out[261]: [5.00000000000000, 6.00000000000000]