我想对一个函数进行傅里叶变换psi(x)，将其乘以 k 空间函数exp(-kx^2-ky^2)，然后将乘积逆傅里叶变换回 x 空间。
但是我的 x 空间和 k 空间网格位于中心，并且我知道我需要fftshift并ifftshift正确实现我的 k 空间乘法。但我不明白它们是如何工作的，所以我不知道按照什么顺序来实现它们。有人可以告诉我我在这里做得是否正确吗？
import scipy.fftpack as spfft
import numpy as np
#Create a centred k-space grid]
kxmax, kymax = 10,10
kxgrid = np.linspace(-kxmax/2, kxmax/2, NX)
kygrid = np.linspace(-kymax/2, kymax/2, NY)
KX, KY = np.meshgrid(kxgrid, kygrid, indexing='xy')
psi = spfft.ifft2(spfft.fftshift(np.exp(-(KX**2 + KY**2)) * spfft.fftshift(spfft.fft2(psi))))