给定x = np.linspace(-3,3,100)并给出函数y = (x+2)*(x-1)*(x-2)，我希望在x时获得y==0的所有值。
我尝试使用x[y==0]，但它只会返回：array([1.])，而我希望得到[-2, 1, 2]。
答案 0 :(得分：1)
试试这个：
from sympy import *
x=Symbol('x')
ans=solve((x+2)*(x-1)*(x-2),x)
print(ans) #output-> [-2,1,2]