python - フーリエ係数の操作

Question

関数の離散フーリエ変換の係数を計算しました。また、方形波の単一サイクルから等間隔に配置された 1000 個のサンプルを含む N=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。