Python中恒定功能的FFT

Question

我想计算恒定信号的DFT（FFT）。这是代码

import matplotlib.pyplot as plt
import numpy as np
from scipy.fftpack import fft, ifft

def constant_function(x):
    return 1

t = np.arange(0.0,1,0.1)
print(type(t1))
print( np.full(t.shape, constant_function(t)))
plt.plot(t, np.full(t.shape, constant_function(t)))

freq = 1
X = fft(constant_function(t))

plt.figure(figsize = (12, 6))
plt.subplot(121)

plt.stem(freq, np.abs(X), 'b', \
         markerfmt=" ", basefmt="-b")
plt.xlabel('Freq (Hz)')
plt.ylabel('FFT Amplitude |X(freq)|')
plt.xlim(0, 1)

plt.subplot(122)
plt.plot(t, ifft(X), 'r')
plt.xlabel('Time (s)')
plt.ylabel('Amplitude')
plt.tight_layout()
plt.show()

，但我会收到此错误：

IndexError: tuple index out of range

我无法以某种方式解决此错误！ 谢谢

Answer 1

问题不是与FFT有关：

def constant_function(x):
   return 1

无论您将其放入什么（您的linspace范围），它都将返回1的常数，并尝试将单个数字的FFT（ 1）而不是数字向量。您可能想要：

def constant_function(x):
    return np.ones(len(x))

尽管在这些情况下，以不同频率创建正弦波的总和通常是测试FFT的最简单方法。