numpy.fft.fft#
- fft.fft(a, n=None, axis=-1, norm=None, out=None)[源码]#
计算一维离散傅里叶变换。
此函数使用高效的快速傅里叶变换 (FFT) 算法 [CT] 计算一维 *n* 点离散傅里叶变换 (DFT)。
- 参数:
- a类数组对象
输入数组,可以是复数。
- nint, optional
输出的变换轴的长度。如果 n 小于输入的长度,则输入将被截断。如果它更大,则输入将被零填充。如果未给出 n,则使用由 axis 指定的轴上的输入长度。
- axisint, optional
计算 FFT 的轴。如果未给出,则使用最后一个轴。
- norm{“backward”, “ortho”, “forward”}, optional
归一化模式(请参阅
numpy.fft)。默认为“backward”。指示前向/后向变换对的哪个方向被缩放以及缩放因子是多少。版本 1.20.0 已添加:添加了“backward”、“forward”值。
- out复数 ndarray,可选
如果提供了此参数,结果将放置在此数组中。它应该具有适当的形状和 dtype。
版本 2.0.0 中新增。
- 返回:
- outcomplex ndarray
在 axis 指定的轴(如果未指定 axis,则为最后一个轴)上进行变换的截断或零填充的输入。
- 引发:
- IndexError
如果 axis 不是 a 的有效轴。
另请参阅
备注
FFT (快速傅里叶变换) 指的是一种计算离散傅里叶变换 (DFT) 的高效方法,通过利用计算项中的对称性。当 n 是 2 的幂时,对称性最高,因此变换对于这些大小的输入最高效。
DFT 的定义,根据本实现中使用的约定,在
numpy.fft模块的文档中。参考
[CT]Cooley, James W., and John W. Tukey, 1965, “An algorithm for the machine calculation of complex Fourier series,” Math. Comput. 19: 297-301。
示例
>>> import numpy as np >>> np.fft.fft(np.exp(2j * np.pi * np.arange(8) / 8)) array([-2.33486982e-16+1.14423775e-17j, 8.00000000e+00-1.25557246e-15j, 2.33486982e-16+2.33486982e-16j, 0.00000000e+00+1.22464680e-16j, -1.14423775e-17+2.33486982e-16j, 0.00000000e+00+5.20784380e-16j, 1.14423775e-17+1.14423775e-17j, 0.00000000e+00+1.22464680e-16j])
在此示例中,实数输入的 FFT 是厄米对称的,即实部对称,虚部反对称,如
numpy.fft文档中所述。>>> import matplotlib.pyplot as plt >>> t = np.arange(256) >>> sp = np.fft.fft(np.sin(t)) >>> freq = np.fft.fftfreq(t.shape[-1]) >>> _ = plt.plot(freq, sp.real, freq, sp.imag) >>> plt.show()