python - 处理傅立叶系数

Question

我已经计算了函数的离散傅里叶变换的系数。我还创建了一个 N=1000 个元素的数组，其中包含来自单个方波周期的 1000 个等距样本。

这是我的代码

 F=zeros(1000)
 F[0:500]=1
 F[500:1000]=-1
 a = rfft(F)
 print a

如何将除前 10 个傅立叶系数之外的所有系数设置为零？

Answer 1

该数组a应包含 501 个复值系数，这些系数从rfft. 相应地，为了只保留前 10 个系数，您可以使用以下方法将剩余系数设置为零：

a[10:500] = 0

或者，您可以使用指定范围一直到最后一个索引a[10:-1] = 0。