python fft算法代码

以下是Python中的FFT算法代码示例：

import numpy as np

# FFT函数
def fft(x):
    N = len(x)
    if N <= 1:
        return x
    else:
        X_even = fft(x[::2])
        X_odd = fft(x[1::2])
        factor = np.exp(-2j * np.pi * np.arange(N) / N)
        return np.concatenate([X_even + factor[:N // 2] * X_odd, X_even + factor[N // 2:] * X_odd])

# 示例
x = np.array([1, 2, 3, 4])
X = fft(x)
print(X)

输出结果：

[10.+0.j -2.+2.j -2.+0.j -2.-2.j]

这表示输入序列经过FFT变换后得到的频域结果。