importmatplotlib.pyplotaspltimportnumpyasnp# Fixing random state for reproducibilitynp.random.seed(19680801)dt=0.01t=np.arange(0,30,dt)nse1=np.random.randn(len(t))# white noise 1nse2=np.random.randn(len(t))# white noise 2# Two signals with a coherent part at 10 Hz and a random parts1=np.sin(2*np.pi*10*t)+nse1s2=np.sin(2*np.pi*10*t)+nse2fig,axs=plt.subplots(2,1,layout='constrained')axs[0].plot(t,s1,t,s2)axs[0].set_xlim(0,2)axs[0].set_xlabel('Time (s)')axs[0].set_ylabel('s1 and s2')axs[0].grid(True)cxy,f=axs[1].cohere(s1,s2,NFFT=256,Fs=1./dt)axs[1].set_ylabel('Coherence')plt.show()

DownloadJupyternotebook:cohere.ipynb

DownloadPythonsourcecode:cohere.py

Downloadzipped:cohere.zip