from scipy.fftpack import fft
import numpy as np
import matplotlib.pyplot as plt


N=1000
Te=1/1500
Te2=1/300
durée = N*Te
durée2=N*Te2

t= np.linspace(0.0,durée,N)
t2=np.linspace(0.0,durée2,N)
f1=50
signal=2*np.cos(2*np.pi*f1*t)+0.5*np.cos(2*np.pi*5*f1*t)
signal2=2*np.cos(2*np.pi*f1*t2)+0.5*np.cos(2*np.pi*5*f1*t2)

fourier=np.fft.fft(signal)
fourier2=np.fft.fft(signal2)
amplitude=2/N*abs(fourier)[:N//2]
amplitude2=2/N*abs(fourier2)[:N//2]
frequence=np.linspace(0,1/(2*Te),N//2)
frequence2=np.linspace(0,1/(2*Te2),N//2)

plt.plot(frequence,amplitude,label='spectre en amplitude avec fe=1500Hz')
plt.grid(True)
plt.legend()
plt.show()
plt.figure()
plt.plot(frequence2,amplitude2,label='spectre en amplitude avec fe=300Hz')
plt.grid(True)
plt.show()
plt.legend()