python - sympy 中求解结果的数值

Question

我试图计算 solve 结果的数值功能。在 WolframAlpha 中，它被称为“近似形式”。
这是我的方程:

>>> solve([Eq(u, r * i), Eq(p, u * i), Eq(p, 20), Eq(r, 4)])
[{i: -√5, p: 20, r: 4, u: -4⋅√5}, {i: √5, p: 20, r: 4, u: 4⋅√5}]

我试过 N ，但它似乎不起作用:

>>> N(solve([Eq(u, r * i), Eq(p, u * i), Eq(p, 20), Eq(r, 4)]), 2)
Traceback (most recent call last):
  File "<string>", line 1, in <module>
  File "/base/data/home/apps/s~sympy-live-hrd/66.426491309333028408/sympy/sympy/core/evalf.py", line 1544, in N
    return sympify(x, rational=True).evalf(n, **options)
AttributeError: 'list' object has no attribute 'evalf'

这样做的正确语法是什么？

Answer 1

N 函数只能应用于 sympy 表达式，不能应用于列表或字典。您可以使用列表/字典推导来重建解决方案数据结构，如下所示:

In [22]: [{k: N(e) for k, e in s.items()} for s in sol]
Out[22]: [{i: -2.23606797749979, p: 20.0, r: 4.0, u: -8.94427190999916}, {i: 2.23606797749979, p: 20.0, r: 4.0, u: 8.94427190999916}]