Laba5

from numpy import array, arange, abs as np_abs
from numpy.fft import rfft, rfftfreq
from numpy.random import uniform
import matplotlib.pyplot as plt
from math import cos,sin,pi

k1=13
k2=27
n=1024
m=10
X=array([8*cos(2*pi*k1*i/n)+11*sin(2*pi*k2*i/n) for i in range(1,n+1)])

spectrum = rfft(X)

FD=1

filtered = list(filter(lambda x: x[1] > 0.01, zip(rfftfreq(n, 1. / FD), np_abs(spectrum) / n)))
print("Составлящие:")
print("     Частота - Амплитуда")
print("\n".join(map(lambda x:str(x[0])+" - "+str(x[1]),filtered)))