In [ ]:
import matplotlib.pyplot as plt
import numpy as np
In [ ]:
Fs = 200.0;  # sampling rate
Ts = 1.0/Fs; # sampling interval
t = np.arange(0,1,Ts) # time vector
In [ ]:
ff = 10;   # frequency of the signal
y = np.sin(2*np.pi*ff*t)
In [ ]:
n = len(y) # length of the signal
k = np.arange(n)
T = n/Fs
frq = k/T # two sides frequency range
frq = frq[range(n//2)] # one side frequency range

Y = np.fft.fft(y)/n # fft computing and normalization
Y = Y[range(n//2)]
In [ ]:
fig, ax = plt.subplots(2, 1, figsize=(10, 8))
ax[0].plot(t,y)
ax[0].set_xlabel('Time')
ax[0].set_ylabel('Amplitude')
ax[1].stem(frq,abs(Y),'r')
ax[1].set_xlabel('Frequency')
ax[1].set_ylabel('Amplitude')
plt.ylim(-1, 1)