操纵傅立叶系数

Question

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

以下是我的代码

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

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

Answer 1

从rfft返回的数组a应该包含501个复值系数。相应地，要仅保留前10个系数，可以使用以下命令将其余系数设置为零：

a[10:500] = 0

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