使用 Sympy 实现用户输入的正确方法？

Question

我目前正在创建一个 python 脚本，它将根据用户输入的公式进行一系列计算；但是，它没有按预期工作？

我尝试了以下方法：

init_printing(use_unicode=True)

x, y = symbols('x y', real = True)
userinput = sympify(input("testinput: "))

x_diff = diff(userinput, x)

print(x_diff)

但是，这总是返回零，但是当我直接写输入时，例如

init_printing(use_unicode=True)

x, y = symbols('x y', real = True)
userinput = x**0.5+y

x_diff = diff(userinput, x)

print(x_diff)

它完美无缺地工作，我在这里做错了什么？

谢谢！

Answer 1

在sympify 函数中添加locals 参数会对您有所帮助。这是一个工作代码，基于你的：

from sympy import *

init_printing(use_unicode=True)

x, y = symbols('x y', real = True)
userinput = input("testinput: ")
locals = {'x':x, 'y':y}
sympified = sympify(userinput, locals=locals)
print(f'derivate /x : = 
 {diff(sympified, x)} 
 derivative / y : 
 {diff(sympified, y)}')

输出：

testinput: cos(y) + 2*x

derivate /x :
2
derivative /y :
-sin(y)