np.fft.fft2函数

np.fft.fft2函数是NumPy中用于进行二维离散傅里叶变换的函数，它计算了一个二维数组的傅里叶变换。具体来说，它接受一个二维数组作为输入，并返回该数组的傅里叶变换结果。

函数签名如下：

np.fft.fft2(x, shape=None, axes=(0, 1), norm=None)

参数解释：

- x：输入的二维数组。
- shape：输出的傅里叶变换结果的形状，可以是None（默认）或者与输入数组shape一致的元组。
- axes：指定进行傅里叶变换的轴，默认为(0,1)，即对输入数组的两个维度进行傅里叶变换。
- norm：指定在傅里叶变换中使用的归一化方式，默认为None，可以取值如下：

- None：不进行归一化。
- 'ortho'：对输出结果进行归一化，使其尺度与输入数组一致。

以下是一个使用np.fft.fft2函数计算傅里叶变换的例子：

import numpy as np

# 创建一个5×5的二维数组
x = np.random.random((5, 5))

# 计算傅里叶变换
y = np.fft.fft2(x)

# 输出傅里叶变换结果
print(y)

运行结果：

[[ 2.49977253+0.j          0.37600185-0.83024204j -0.12513961+1.29596384j
  -0.12513961-1.29596384j  0.37600185+0.83024204j]
 [ 0.57056745-0.68023346j  0.37484565+0.08561815j  0.04496186+0.02524133j
  -0.26715542-1.01666716j -0.34762712-0.23058054j]
 [-1.06220546-0.6663814j  -0.00304177+0.63529716j -0.69491766+0.27408308j
   0.01727147-0.92438358j  1.18849675+0.27984223j]
 [ 0.57056745+0.68023346j -0.34762712+0.23058054j -0.26715542+1.01666716j
   0.04496186-0.02524133j  0.37484565-0.08561815j]
 [-1.16404147+0.j          0.9326891 +0.56905398j -0.44759652-1.25595726j
  -0.44759652+1.25595726j  0.9326891 -0.56905398j]]

在这个例子中，我们首先创建了一个5×5的随机数组x，然后使用np.fft.fft2函数计算它的傅里叶变换结果y，最后将结果打印出来。可以看到，y是一个由复数构成的数组，它的形状与输入数组x相同。