二维数组使用np.fft.fft2()函数进行FFT变换，怎么画出k空间

要画出二维数组的k空间，可以使用np.fft.fftfreq()函数获取kx和ky的值，然后将其绘制成网格图。

下面是一个示例代码：

import numpy as np
import matplotlib.pyplot as plt

# 生成一个二维正弦函数
x = np.linspace(-10, 10, 101)
y = np.linspace(-10, 10, 101)
xx, yy = np.meshgrid(x, y)
f = np.sin(xx) * np.cos(yy)

# 进行FFT变换
F = np.fft.fft2(f)
F_shift = np.fft.fftshift(F)

# 获取kx和ky的值
kx = np.fft.fftfreq(F_shift.shape[0])
ky = np.fft.fftfreq(F_shift.shape[1])
kx, ky = np.meshgrid(kx, ky)

# 绘制网格图
plt.figure()
plt.pcolormesh(kx, ky, np.abs(F_shift))
plt.xlabel('kx')
plt.ylabel('ky')
plt.colorbar()
plt.show()

这段代码首先生成一个二维正弦函数，然后对其进行FFT变换，并将结果进行平移以使得直流分量位于中心。接着，使用np.fft.fftfreq()函数获取kx和ky的值，并将其绘制成网格图。最后使用plt.colorbar()函数添加颜色条。

运行代码后，就可以得到k空间网格图，其中x轴和y轴为kx和ky的值，颜色表示对应的振幅大小。