python fft横轴_IFFT-FFT后缩放X轴

importnumpyasnpimportmatplotlib.pyplotaspltimportscipyfromscipy.interpolateimportinterp1dC_LIGHT=299.792# for easier case, this is zero, so it can be ignored.def_disp(x,GD=0,GDD=0,TOD=0,FOD=0,QOD=0):returnx*GD+(GDD/2)*x**2+(TOD/6)*x**3+(FOD/24)*x**4+(QOD/120)*x**5# the generator to make sample datasetsdefgenerator(start,stop,center,delay,GD=0,GDD=0,TOD=0,FOD=0,QOD=0,resolution=0.1,pulse_duration=15,chirp=0):window=(np.sqrt(1+chirp**2)*8*np.log(2))/(pulse_duration**2)lamend=(2*np.pi*C_LIGHT)/startlamstart=(2*np.pi*C_LIGHT)/stoplam=np.arange(lamstart,lamend+resolution,resolution)omega=(2*np.pi*C_LIGHT)/lamrelom=omega-centeri_r=np.exp(-(relom)**2/(window))i_s=np.exp(-(relom)**2/(window))i=i_r+i_s+2*np.sqrt(i_r*i_s)*np.cos(_disp(relom,GD=GD,GDD=GDD,TOD=TOD,FOD=FOD,QOD=QOD)+delay*omega)#since the _disp polynomial is set to be zero, it's just cos(delay*omega)returnomega,idefinterpol(x,y):''' Simple linear interpolation '''xs=np.linspace(x[0],x[-1],len(x))intp=interp1d(x,y,kind='linear',fill_value='extrapolate')ys=intp(xs)returnxs,ysdefifft_method(initSpectrumX,initSpectrumY,interpolate=True):iflen(initSpectrumY)>0andlen(initSpectrumX)>0:Ydata=initSpectrumYXdata=initSpectrumXelse:raiseValueErrorN=len(Xdata)ifinterpolate:Xdata,Ydata=interpol(Xdata,Ydata)# the (2*np.pi) division is because we have angular frequency, not frequencyxf=np.fft.fftfreq(N,d=(Xdata[1]-Xdata[0])/(2*np.pi))*N*Xdata[-1]/(N-1)yf=np.fft.ifft(Ydata)else:pass# some irrelevant code therereturnxf,yfdeffft_method(initSpectrumX,initSpectrumY):iflen(initSpectrumY)>0andlen(initSpectrumX)>0:Ydata=initSpectrumYXdata=initSpectrumXelse:raiseValueErroryf=np.fft.fft(Ydata)xf=np.fft.fftfreq(len(Xdata),d=(Xdata[1]-Xdata[0])*2*np.pi)# the problem is there, where I transform the x values.xf=np.fft.ifftshift(xf)returnxf,yf# the generated datax,y=generator(1,3,2,delay=1500,resolution=0.1)# plt.plot(x,y)xx,yy=ifft_method(x,y)#if the x values are correctly scaled, the two symmetrical spikes should appear exactly at delay value# plt.plot(xx, np.abs(yy))#do the cutting there, which is also irrelevant now# the problem is there, in fft_method. The x values are not the same as before transforms.xxx,yyy=fft_method(xx,yy)plt.plot(xxx,np.abs(yyy))#and it should look like this:#xs = np.linspace(x[0], x[-1], len(x))#plt.plot(xs, np.abs(yyy))plt.grid()plt.show()