离散傅里叶变换python
时间: 2023-10-15 22:22:52 浏览: 113
离散傅里叶变换
离散傅里叶变换(Discrete Fourier Transform,DFT)在Python中可以使用NumPy库来实现。以下是一个示例代码:
```python
import numpy as np
# 定义离散傅里叶变换函数
def dft(x):
N = len(x)
n = np.arange(N)
k = n.reshape((N, 1))
e = np.exp(-2j * np.pi * k * n / N)
return np.dot(e, x)
# 输入信号
x = np.array([1, 2, 3, 4])
# 调用离散傅里叶变换函数
X = dft(x)
# 输出频谱
print(X)
```
运行以上代码,将会输出信号的离散傅里叶变换结果。
请注意,这只是一个简单的示例代码,实际应用中可能需要对输入信号进行零填充(zero-padding)等处理。此外,在Python中也可以使用SciPy库提供的`fft`函数来实现离散傅里叶变换,该函数在处理效率上更高。
阅读全文