Compute Cross Spectral Density in Python

Introduction

In signal processing, Cross Spectral Density (CSD) is a measure of the correlation between two signals in the frequency domain. It is used to determine how much two signals are related to each other in terms of their frequency content. In this lab, you will learn how to compute the CSD of two signals using Python's Matplotlib library.

VM Tips

After the VM startup is done, click the top left corner to switch to the Notebook tab to access Jupyter Notebook for practice.

Sometimes, you may need to wait a few seconds for Jupyter Notebook to finish loading. The validation of operations cannot be automated because of limitations in Jupyter Notebook.

If you face issues during learning, feel free to ask Labby. Provide feedback after the session, and we will promptly resolve the problem for you.

Skills Graph

%%%%{init: {'theme':'neutral'}}%%%% flowchart RL python(("`Python`")) -.-> python/BasicConceptsGroup(["`Basic Concepts`"]) python(("`Python`")) -.-> python/FileHandlingGroup(["`File Handling`"]) matplotlib(("`Matplotlib`")) -.-> matplotlib/BasicConceptsGroup(["`Basic Concepts`"]) matplotlib(("`Matplotlib`")) -.-> matplotlib/PlottingDataGroup(["`Plotting Data`"]) matplotlib(("`Matplotlib`")) -.-> matplotlib/PlotCustomizationGroup(["`Plot Customization`"]) python(("`Python`")) -.-> python/ControlFlowGroup(["`Control Flow`"]) python(("`Python`")) -.-> python/DataStructuresGroup(["`Data Structures`"]) python(("`Python`")) -.-> python/ModulesandPackagesGroup(["`Modules and Packages`"]) python(("`Python`")) -.-> python/PythonStandardLibraryGroup(["`Python Standard Library`"]) python(("`Python`")) -.-> python/DataScienceandMachineLearningGroup(["`Data Science and Machine Learning`"]) python(("`Python`")) -.-> python/FunctionsGroup(["`Functions`"]) python/BasicConceptsGroup -.-> python/comments("`Comments`") python/FileHandlingGroup -.-> python/with_statement("`Using with Statement`") matplotlib/BasicConceptsGroup -.-> matplotlib/importing_matplotlib("`Importing Matplotlib`") matplotlib/BasicConceptsGroup -.-> matplotlib/figures_axes("`Understanding Figures and Axes`") matplotlib/PlottingDataGroup -.-> matplotlib/line_plots("`Line Plots`") matplotlib/PlotCustomizationGroup -.-> matplotlib/legend_config("`Legend Configuration`") python/ControlFlowGroup -.-> python/for_loops("`For Loops`") python/DataStructuresGroup -.-> python/tuples("`Tuples`") python/ModulesandPackagesGroup -.-> python/importing_modules("`Importing Modules`") python/ModulesandPackagesGroup -.-> python/standard_libraries("`Common Standard Libraries`") python/PythonStandardLibraryGroup -.-> python/math_random("`Math and Random`") python/DataScienceandMachineLearningGroup -.-> python/numerical_computing("`Numerical Computing`") python/DataScienceandMachineLearningGroup -.-> python/data_visualization("`Data Visualization`") python/FunctionsGroup -.-> python/build_in_functions("`Build-in Functions`") subgraph Lab Skills python/comments -.-> lab-48634{{"`Compute Cross Spectral Density in Python`"}} python/with_statement -.-> lab-48634{{"`Compute Cross Spectral Density in Python`"}} matplotlib/importing_matplotlib -.-> lab-48634{{"`Compute Cross Spectral Density in Python`"}} matplotlib/figures_axes -.-> lab-48634{{"`Compute Cross Spectral Density in Python`"}} matplotlib/line_plots -.-> lab-48634{{"`Compute Cross Spectral Density in Python`"}} matplotlib/legend_config -.-> lab-48634{{"`Compute Cross Spectral Density in Python`"}} python/for_loops -.-> lab-48634{{"`Compute Cross Spectral Density in Python`"}} python/tuples -.-> lab-48634{{"`Compute Cross Spectral Density in Python`"}} python/importing_modules -.-> lab-48634{{"`Compute Cross Spectral Density in Python`"}} python/standard_libraries -.-> lab-48634{{"`Compute Cross Spectral Density in Python`"}} python/math_random -.-> lab-48634{{"`Compute Cross Spectral Density in Python`"}} python/numerical_computing -.-> lab-48634{{"`Compute Cross Spectral Density in Python`"}} python/data_visualization -.-> lab-48634{{"`Compute Cross Spectral Density in Python`"}} python/build_in_functions -.-> lab-48634{{"`Compute Cross Spectral Density in Python`"}} end

Import Required Libraries

We need to import the following libraries: numpy and matplotlib.pyplot.

import matplotlib.pyplot as plt
import numpy as np

Generate Signals

We need to generate two signals. These signals contain a coherent part and a random part. The coherent part of both signals has a frequency of 10 Hz. The random part of the signals is generated using white noise that is passed through a low-pass filter to create colored noise.

dt = 0.01
t = np.arange(0, 30, dt)

## Fixing random state for reproducibility
np.random.seed(19680801)

nse1 = np.random.randn(len(t))                 ## white noise 1
nse2 = np.random.randn(len(t))                 ## white noise 2
r = np.exp(-t / 0.05)

cnse1 = np.convolve(nse1, r, mode='same') * dt   ## colored noise 1
cnse2 = np.convolve(nse2, r, mode='same') * dt   ## colored noise 2

## two signals with a coherent part and a random part
s1 = 0.01 * np.sin(2 * np.pi * 10 * t) + cnse1
s2 = 0.01 * np.sin(2 * np.pi * 10 * t) + cnse2

Plot Signals

We can plot the two generated signals using Matplotlib's plot function.

fig, ax = plt.subplots()
ax.plot(t, s1, label='s1')
ax.plot(t, s2, label='s2')
ax.set_xlabel('Time')
ax.set_ylabel('Amplitude')
ax.legend()
plt.show()

Compute CSD

To compute the CSD of two signals, we need to use Matplotlib's csd function. The function takes the two signals, the number of points for the FFT, and the sampling frequency as inputs.

fig, ax = plt.subplots()
cxy, f = ax.csd(s1, s2, 256, 1. / dt)
ax.set_ylabel('CSD (dB)')
plt.show()

Interpret Results

The resulting plot shows the CSD of the two signals. The x-axis represents the frequency and the y-axis represents the strength of the correlation between the two signals at that frequency. The plot shows a peak at 10 Hz, which is the frequency of the coherent part of the signals. This indicates that the two signals are strongly correlated at this frequency.

Summary

In this lab, you learned how to compute the Cross Spectral Density of two signals using Python's Matplotlib library. We generated two signals with a coherent part and a random part, plotted the signals, computed the CSD, and interpreted the results. The CSD is a useful tool for determining the correlation between two signals in the frequency domain.