我正在使用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或者其他东西,但那看起来很难看,也许有内置的方法
答案 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]