kw = np.fft.ifftshift(kw)