我已经计算了函数的离散傅立叶变换的系数。我还创建了一个N = 1000个元素的数组,其中包含来自方波的单个周期的1000个等间隔样本。
这是我的代码
F=zeros(1000)
F[0:500]=1
F[500:1000]=-1
a = rfft(F)
print a
如何将除前10个傅立叶系数之外的所有系数设置为零?
答案 0 :(得分:0)
数组a
应包含从rfft
返回的501个复数值系数。相应地只保留前10个系数,您可以将剩余系数设置为零:
a[10:500] = 0
或者,您可以使用a[10:-1] = 0
指定一直到最后一个索引的范围。