操纵傅里叶系数

时间:2015-12-06 19:56:29

标签: python arrays fft coefficients

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

这是我的代码

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

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

1 个答案:

答案 0 :(得分:0)

数组a应包含从rfft返回的501个复数值系数。相应地只保留前10个系数,您可以将剩余系数设置为零:

a[10:500] = 0

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