将sympy符号变量转换为numpy数组

时间:2019-07-01 07:40:31

标签: python numpy sympy convolution

我想执行一个包含sympy符号变量的卷积,然后将其转换为numpy数组。

我的MWE是:

from numpy import pi, float64, linspace
from scipy.signal import fftconvolve
import matplotlib.pyplot as plt
from sympy import symbols
from sympy.utilities.lambdify import lambdify

a = 0.657
b = 0.745
c = 0.642
d = 0.343

x = symbols('x')
f = 2*b / ((x-a)**2 + b**2)
g = 2*d / ((x-c)**2 + d**2)
fog = fftconvolve(f,g,mode='same')

fog_fun = lambdify(x,fog,'numpy') # returns a numpy-ready function
xlist = linspace(-20,20,int(1e3))
my_fog = fog_fun(xlist)

dx = xlist[1]-xlist[0]

fog1 = 4*pi*(b+d)/((x-a-c)**2+(b+d)**2) # correct analytic solution

plt.figure()
plt.plot(x,fog1,lw=2,label='analytic')
plt.plot(x,my_fog*dx,lw=2,label='sympy')
plt.grid()
plt.legend(loc='best')
plt.show()

我尝试使用建议的解决方案here,但出现错误TypeError: can't convert expression to float。我不确定该如何解决。

(注意:这是MWE。我实际使用的实际fg比本文中定义的Lorentzians复杂得多。)

1 个答案:

答案 0 :(得分:1)

当您试图绘制符号项时,出现的错误是符号线,

plt.plot(x,fog1,lw=2,label='analytic')

如果您对my_fog使用转换后的xlist

plt.plot(xlist,my_fog*dx,lw=2,label='sympy')

它看起来像是洛伦兹分布,

enter image description here