Compute a cross-spectral density (CSD) matrix#

A cross-spectral density (CSD) matrix is similar to a covariance matrix, but in the time-frequency domain. It is the first step towards computing sensor-to-sensor coherence or a DICS beamformer.

This script demonstrates the three methods that MNE-Python provides to compute the CSD:

  1. Using short-term Fourier transform: mne.time_frequency.csd_fourier()

  2. Using a multitaper approach: mne.time_frequency.csd_multitaper()

  3. Using Morlet wavelets: mne.time_frequency.csd_morlet()

# Author: Marijn van Vliet <w.m.vanvliet@gmail.com>
# License: BSD-3-Clause
# Copyright the MNE-Python contributors.
import mne
from mne.datasets import sample
from mne.time_frequency import csd_fourier, csd_morlet, csd_multitaper

print(__doc__)

In the following example, the computation of the CSD matrices can be performed using multiple cores. Set n_jobs to a value >1 to select the number of cores to use.

n_jobs = 1

Loading the sample dataset.

data_path = sample.data_path()
meg_path = data_path / "MEG" / "sample"
fname_raw = meg_path / "sample_audvis_raw.fif"
fname_event = meg_path / "sample_audvis_raw-eve.fif"
raw = mne.io.read_raw_fif(fname_raw)
events = mne.read_events(fname_event)
Opening raw data file /home/circleci/mne_data/MNE-sample-data/MEG/sample/sample_audvis_raw.fif...
    Read a total of 3 projection items:
        PCA-v1 (1 x 102)  idle
        PCA-v2 (1 x 102)  idle
        PCA-v3 (1 x 102)  idle
    Range : 25800 ... 192599 =     42.956 ...   320.670 secs
Ready.

By default, CSD matrices are computed using all MEG/EEG channels. When interpreting a CSD matrix with mixed sensor types, be aware that the measurement units, and thus the scalings, differ across sensors. In this example, for speed and clarity, we select a single channel type: gradiometers.

picks = mne.pick_types(raw.info, meg="grad")

# Make some epochs, based on events with trigger code 1
epochs = mne.Epochs(
    raw,
    events,
    event_id=1,
    tmin=-0.2,
    tmax=1,
    picks=picks,
    baseline=(None, 0),
    reject=dict(grad=4000e-13),
    preload=True,
)
Not setting metadata
72 matching events found
Setting baseline interval to [-0.19979521315838786, 0.0] s
Applying baseline correction (mode: mean)
0 projection items activated
Loading data for 72 events and 722 original time points ...
0 bad epochs dropped

Computing CSD matrices using short-term Fourier transform and (adaptive) multitapers is straightforward:

csd_fft = csd_fourier(epochs, fmin=15, fmax=20, n_jobs=n_jobs)
csd_mt = csd_multitaper(epochs, fmin=15, fmax=20, adaptive=True, n_jobs=n_jobs)
Computing cross-spectral density from epochs...

  0%|          | CSD epoch blocks : 0/72 [00:00<?,       ?it/s]
  1%|▏         | CSD epoch blocks : 1/72 [00:00<00:02,   24.18it/s]
  3%|▎         | CSD epoch blocks : 2/72 [00:00<00:02,   32.06it/s]
  4%|▍         | CSD epoch blocks : 3/72 [00:00<00:01,   37.37it/s]
  6%|▌         | CSD epoch blocks : 4/72 [00:00<00:01,   41.64it/s]
  7%|▋         | CSD epoch blocks : 5/72 [00:00<00:01,   44.69it/s]
  8%|▊         | CSD epoch blocks : 6/72 [00:00<00:01,   46.99it/s]
 10%|▉         | CSD epoch blocks : 7/72 [00:00<00:01,   48.83it/s]
 11%|█         | CSD epoch blocks : 8/72 [00:00<00:01,   50.17it/s]
 12%|█▎        | CSD epoch blocks : 9/72 [00:00<00:01,   51.19it/s]
 14%|█▍        | CSD epoch blocks : 10/72 [00:00<00:01,   52.14it/s]
 15%|█▌        | CSD epoch blocks : 11/72 [00:00<00:01,   52.99it/s]
 17%|█▋        | CSD epoch blocks : 12/72 [00:00<00:01,   53.66it/s]
 18%|█▊        | CSD epoch blocks : 13/72 [00:00<00:01,   54.25it/s]
 19%|█▉        | CSD epoch blocks : 14/72 [00:00<00:01,   54.88it/s]
 21%|██        | CSD epoch blocks : 15/72 [00:00<00:01,   55.36it/s]
 22%|██▏       | CSD epoch blocks : 16/72 [00:00<00:01,   55.76it/s]
 24%|██▎       | CSD epoch blocks : 17/72 [00:00<00:00,   55.36it/s]
 25%|██▌       | CSD epoch blocks : 18/72 [00:00<00:00,   54.35it/s]
 26%|██▋       | CSD epoch blocks : 19/72 [00:00<00:00,   53.49it/s]
 28%|██▊       | CSD epoch blocks : 20/72 [00:00<00:00,   53.32it/s]
 29%|██▉       | CSD epoch blocks : 21/72 [00:00<00:00,   53.80it/s]
 31%|███       | CSD epoch blocks : 22/72 [00:00<00:00,   54.24it/s]
 32%|███▏      | CSD epoch blocks : 23/72 [00:00<00:00,   54.59it/s]
 33%|███▎      | CSD epoch blocks : 24/72 [00:00<00:00,   54.95it/s]
 35%|███▍      | CSD epoch blocks : 25/72 [00:00<00:00,   55.25it/s]
 36%|███▌      | CSD epoch blocks : 26/72 [00:00<00:00,   55.56it/s]
 38%|███▊      | CSD epoch blocks : 27/72 [00:00<00:00,   55.64it/s]
 39%|███▉      | CSD epoch blocks : 28/72 [00:00<00:00,   55.95it/s]
 40%|████      | CSD epoch blocks : 29/72 [00:00<00:00,   56.20it/s]
 42%|████▏     | CSD epoch blocks : 30/72 [00:00<00:00,   56.41it/s]
 43%|████▎     | CSD epoch blocks : 31/72 [00:00<00:00,   56.62it/s]
 44%|████▍     | CSD epoch blocks : 32/72 [00:00<00:00,   56.86it/s]
 46%|████▌     | CSD epoch blocks : 33/72 [00:00<00:00,   57.00it/s]
 47%|████▋     | CSD epoch blocks : 34/72 [00:00<00:00,   57.23it/s]
 49%|████▊     | CSD epoch blocks : 35/72 [00:00<00:00,   57.47it/s]
 50%|█████     | CSD epoch blocks : 36/72 [00:00<00:00,   57.64it/s]
 51%|█████▏    | CSD epoch blocks : 37/72 [00:00<00:00,   57.68it/s]
 53%|█████▎    | CSD epoch blocks : 38/72 [00:00<00:00,   57.59it/s]
 54%|█████▍    | CSD epoch blocks : 39/72 [00:00<00:00,   56.65it/s]
 56%|█████▌    | CSD epoch blocks : 40/72 [00:00<00:00,   56.89it/s]
 57%|█████▋    | CSD epoch blocks : 41/72 [00:00<00:00,   57.08it/s]
 58%|█████▊    | CSD epoch blocks : 42/72 [00:00<00:00,   57.26it/s]
 60%|█████▉    | CSD epoch blocks : 43/72 [00:00<00:00,   57.41it/s]
 61%|██████    | CSD epoch blocks : 44/72 [00:00<00:00,   57.59it/s]
 62%|██████▎   | CSD epoch blocks : 45/72 [00:00<00:00,   56.17it/s]
 64%|██████▍   | CSD epoch blocks : 46/72 [00:00<00:00,   54.99it/s]
 65%|██████▌   | CSD epoch blocks : 47/72 [00:00<00:00,   54.60it/s]
 67%|██████▋   | CSD epoch blocks : 48/72 [00:00<00:00,   54.90it/s]
 68%|██████▊   | CSD epoch blocks : 49/72 [00:00<00:00,   55.24it/s]
 69%|██████▉   | CSD epoch blocks : 50/72 [00:00<00:00,   55.44it/s]
 71%|███████   | CSD epoch blocks : 51/72 [00:00<00:00,   55.73it/s]
 72%|███████▏  | CSD epoch blocks : 52/72 [00:00<00:00,   55.84it/s]
 74%|███████▎  | CSD epoch blocks : 53/72 [00:00<00:00,   56.05it/s]
 75%|███████▌  | CSD epoch blocks : 54/72 [00:00<00:00,   56.26it/s]
 76%|███████▋  | CSD epoch blocks : 55/72 [00:00<00:00,   56.49it/s]
 78%|███████▊  | CSD epoch blocks : 56/72 [00:01<00:00,   56.71it/s]
 79%|███████▉  | CSD epoch blocks : 57/72 [00:01<00:00,   56.87it/s]
 81%|████████  | CSD epoch blocks : 58/72 [00:01<00:00,   57.12it/s]
 82%|████████▏ | CSD epoch blocks : 59/72 [00:01<00:00,   57.31it/s]
 83%|████████▎ | CSD epoch blocks : 60/72 [00:01<00:00,   57.51it/s]
 85%|████████▍ | CSD epoch blocks : 61/72 [00:01<00:00,   57.73it/s]
 86%|████████▌ | CSD epoch blocks : 62/72 [00:01<00:00,   57.88it/s]
 88%|████████▊ | CSD epoch blocks : 63/72 [00:01<00:00,   57.94it/s]
 89%|████████▉ | CSD epoch blocks : 64/72 [00:01<00:00,   58.14it/s]
 90%|█████████ | CSD epoch blocks : 65/72 [00:01<00:00,   58.24it/s]
 92%|█████████▏| CSD epoch blocks : 66/72 [00:01<00:00,   58.30it/s]
 93%|█████████▎| CSD epoch blocks : 67/72 [00:01<00:00,   58.39it/s]
 94%|█████████▍| CSD epoch blocks : 68/72 [00:01<00:00,   58.54it/s]
 96%|█████████▌| CSD epoch blocks : 69/72 [00:01<00:00,   58.62it/s]
 97%|█████████▋| CSD epoch blocks : 70/72 [00:01<00:00,   58.69it/s]
 99%|█████████▊| CSD epoch blocks : 71/72 [00:01<00:00,   58.85it/s]
100%|██████████| CSD epoch blocks : 72/72 [00:01<00:00,   58.87it/s]
100%|██████████| CSD epoch blocks : 72/72 [00:01<00:00,   56.65it/s]
[done]
    Using multitaper spectrum estimation with 7 DPSS windows
Computing cross-spectral density from epochs...

  0%|          | CSD epoch blocks : 0/72 [00:00<?,       ?it/s]
  1%|▏         | CSD epoch blocks : 1/72 [00:00<00:14,    5.05it/s]
  3%|▎         | CSD epoch blocks : 2/72 [00:00<00:11,    5.87it/s]
  4%|▍         | CSD epoch blocks : 3/72 [00:00<00:11,    6.25it/s]
  6%|▌         | CSD epoch blocks : 4/72 [00:00<00:10,    6.47it/s]
  7%|▋         | CSD epoch blocks : 5/72 [00:00<00:10,    6.40it/s]
  8%|▊         | CSD epoch blocks : 6/72 [00:00<00:10,    6.53it/s]
 10%|▉         | CSD epoch blocks : 7/72 [00:01<00:09,    6.63it/s]
 11%|█         | CSD epoch blocks : 8/72 [00:01<00:09,    6.53it/s]
 12%|█▎        | CSD epoch blocks : 9/72 [00:01<00:09,    6.59it/s]
 14%|█▍        | CSD epoch blocks : 10/72 [00:01<00:09,    6.60it/s]
 15%|█▌        | CSD epoch blocks : 11/72 [00:01<00:09,    6.50it/s]
 17%|█▋        | CSD epoch blocks : 12/72 [00:01<00:09,    6.57it/s]
 18%|█▊        | CSD epoch blocks : 13/72 [00:01<00:08,    6.62it/s]
 19%|█▉        | CSD epoch blocks : 14/72 [00:02<00:08,    6.67it/s]
 21%|██        | CSD epoch blocks : 15/72 [00:02<00:08,    6.70it/s]
 22%|██▏       | CSD epoch blocks : 16/72 [00:02<00:08,    6.73it/s]
 24%|██▎       | CSD epoch blocks : 17/72 [00:02<00:08,    6.71it/s]
 25%|██▌       | CSD epoch blocks : 18/72 [00:02<00:08,    6.75it/s]
 26%|██▋       | CSD epoch blocks : 19/72 [00:02<00:07,    6.71it/s]
 28%|██▊       | CSD epoch blocks : 20/72 [00:02<00:07,    6.73it/s]
 29%|██▉       | CSD epoch blocks : 21/72 [00:03<00:07,    6.74it/s]
 31%|███       | CSD epoch blocks : 22/72 [00:03<00:07,    6.75it/s]
 32%|███▏      | CSD epoch blocks : 23/72 [00:03<00:07,    6.77it/s]
 33%|███▎      | CSD epoch blocks : 24/72 [00:03<00:07,    6.78it/s]
 35%|███▍      | CSD epoch blocks : 25/72 [00:03<00:06,    6.81it/s]
 36%|███▌      | CSD epoch blocks : 26/72 [00:03<00:06,    6.78it/s]
 38%|███▊      | CSD epoch blocks : 27/72 [00:04<00:06,    6.73it/s]
 39%|███▉      | CSD epoch blocks : 28/72 [00:04<00:06,    6.76it/s]
 40%|████      | CSD epoch blocks : 29/72 [00:04<00:06,    6.80it/s]
 42%|████▏     | CSD epoch blocks : 30/72 [00:04<00:06,    6.82it/s]
 43%|████▎     | CSD epoch blocks : 31/72 [00:04<00:06,    6.79it/s]
 44%|████▍     | CSD epoch blocks : 32/72 [00:04<00:05,    6.81it/s]
 46%|████▌     | CSD epoch blocks : 33/72 [00:04<00:05,    6.83it/s]
 47%|████▋     | CSD epoch blocks : 34/72 [00:05<00:05,    6.82it/s]
 49%|████▊     | CSD epoch blocks : 35/72 [00:05<00:05,    6.80it/s]
 50%|█████     | CSD epoch blocks : 36/72 [00:05<00:05,    6.82it/s]
 51%|█████▏    | CSD epoch blocks : 37/72 [00:05<00:05,    6.83it/s]
 53%|█████▎    | CSD epoch blocks : 38/72 [00:05<00:04,    6.84it/s]
 54%|█████▍    | CSD epoch blocks : 39/72 [00:05<00:04,    6.86it/s]
 56%|█████▌    | CSD epoch blocks : 40/72 [00:05<00:04,    6.87it/s]
 57%|█████▋    | CSD epoch blocks : 41/72 [00:06<00:04,    6.89it/s]
 58%|█████▊    | CSD epoch blocks : 42/72 [00:06<00:04,    6.90it/s]
 60%|█████▉    | CSD epoch blocks : 43/72 [00:06<00:04,    6.80it/s]
 61%|██████    | CSD epoch blocks : 44/72 [00:06<00:04,    6.77it/s]
 62%|██████▎   | CSD epoch blocks : 45/72 [00:06<00:04,    6.74it/s]
 64%|██████▍   | CSD epoch blocks : 46/72 [00:06<00:03,    6.68it/s]
 65%|██████▌   | CSD epoch blocks : 47/72 [00:06<00:03,    6.68it/s]
 67%|██████▋   | CSD epoch blocks : 48/72 [00:07<00:03,    6.70it/s]
 68%|██████▊   | CSD epoch blocks : 49/72 [00:07<00:03,    6.72it/s]
 69%|██████▉   | CSD epoch blocks : 50/72 [00:07<00:03,    6.73it/s]
 71%|███████   | CSD epoch blocks : 51/72 [00:07<00:03,    6.75it/s]
 72%|███████▏  | CSD epoch blocks : 52/72 [00:07<00:02,    6.76it/s]
 74%|███████▎  | CSD epoch blocks : 53/72 [00:07<00:02,    6.77it/s]
 75%|███████▌  | CSD epoch blocks : 54/72 [00:07<00:02,    6.79it/s]
 76%|███████▋  | CSD epoch blocks : 55/72 [00:08<00:02,    6.79it/s]
 78%|███████▊  | CSD epoch blocks : 56/72 [00:08<00:02,    6.80it/s]
 79%|███████▉  | CSD epoch blocks : 57/72 [00:08<00:02,    6.76it/s]
 81%|████████  | CSD epoch blocks : 58/72 [00:08<00:02,    6.78it/s]
 82%|████████▏ | CSD epoch blocks : 59/72 [00:08<00:01,    6.75it/s]
 83%|████████▎ | CSD epoch blocks : 60/72 [00:08<00:01,    6.76it/s]
 85%|████████▍ | CSD epoch blocks : 61/72 [00:09<00:01,    6.77it/s]
 86%|████████▌ | CSD epoch blocks : 62/72 [00:09<00:01,    6.79it/s]
 88%|████████▊ | CSD epoch blocks : 63/72 [00:09<00:01,    6.81it/s]
 89%|████████▉ | CSD epoch blocks : 64/72 [00:09<00:01,    6.83it/s]
 90%|█████████ | CSD epoch blocks : 65/72 [00:09<00:01,    6.83it/s]
 92%|█████████▏| CSD epoch blocks : 66/72 [00:09<00:00,    6.85it/s]
 93%|█████████▎| CSD epoch blocks : 67/72 [00:09<00:00,    6.84it/s]
 94%|█████████▍| CSD epoch blocks : 68/72 [00:10<00:00,    6.85it/s]
 96%|█████████▌| CSD epoch blocks : 69/72 [00:10<00:00,    6.84it/s]
 97%|█████████▋| CSD epoch blocks : 70/72 [00:10<00:00,    6.82it/s]
 99%|█████████▊| CSD epoch blocks : 71/72 [00:10<00:00,    6.82it/s]
100%|██████████| CSD epoch blocks : 72/72 [00:10<00:00,    6.83it/s]
100%|██████████| CSD epoch blocks : 72/72 [00:10<00:00,    6.78it/s]
[done]

When computing the CSD with Morlet wavelets, you specify the exact frequencies at which to compute it. For each frequency, a corresponding wavelet will be constructed and convolved with the signal, resulting in a time-frequency decomposition.

The CSD is constructed by computing the correlation between the time-frequency representations between all sensor-to-sensor pairs. The time-frequency decomposition originally has the same sampling rate as the signal, in our case ~600Hz. This means the decomposition is over-specified in time and we may not need to use all samples during our CSD computation, just enough to get a reliable correlation statistic. By specifying decim=10, we use every 10th sample, which will greatly speed up the computation and will have a minimal effect on the CSD.

frequencies = [16, 17, 18, 19, 20]
csd_wav = csd_morlet(epochs, frequencies, decim=10, n_jobs=n_jobs)
Computing cross-spectral density from epochs...

  0%|          | CSD epoch blocks : 0/72 [00:00<?,       ?it/s]
  1%|▏         | CSD epoch blocks : 1/72 [00:00<00:07,    9.21it/s]
  3%|▎         | CSD epoch blocks : 2/72 [00:00<00:06,   11.14it/s]
  4%|▍         | CSD epoch blocks : 3/72 [00:00<00:05,   11.59it/s]
  6%|▌         | CSD epoch blocks : 4/72 [00:00<00:05,   12.20it/s]
  7%|▋         | CSD epoch blocks : 5/72 [00:00<00:05,   12.57it/s]
  8%|▊         | CSD epoch blocks : 6/72 [00:00<00:05,   12.84it/s]
 10%|▉         | CSD epoch blocks : 7/72 [00:00<00:04,   13.02it/s]
 11%|█         | CSD epoch blocks : 8/72 [00:00<00:04,   13.15it/s]
 12%|█▎        | CSD epoch blocks : 9/72 [00:00<00:04,   12.97it/s]
 14%|█▍        | CSD epoch blocks : 10/72 [00:00<00:04,   12.96it/s]
 15%|█▌        | CSD epoch blocks : 11/72 [00:00<00:04,   13.03it/s]
 17%|█▋        | CSD epoch blocks : 12/72 [00:00<00:04,   13.14it/s]
 18%|█▊        | CSD epoch blocks : 13/72 [00:00<00:04,   13.16it/s]
 19%|█▉        | CSD epoch blocks : 14/72 [00:01<00:04,   13.15it/s]
 21%|██        | CSD epoch blocks : 15/72 [00:01<00:04,   13.21it/s]
 22%|██▏       | CSD epoch blocks : 16/72 [00:01<00:04,   13.29it/s]
 24%|██▎       | CSD epoch blocks : 17/72 [00:01<00:04,   13.31it/s]
 25%|██▌       | CSD epoch blocks : 18/72 [00:01<00:04,   13.36it/s]
 26%|██▋       | CSD epoch blocks : 19/72 [00:01<00:04,   13.16it/s]
 28%|██▊       | CSD epoch blocks : 20/72 [00:01<00:03,   13.21it/s]
 29%|██▉       | CSD epoch blocks : 21/72 [00:01<00:03,   13.12it/s]
 31%|███       | CSD epoch blocks : 22/72 [00:01<00:03,   12.92it/s]
 32%|███▏      | CSD epoch blocks : 23/72 [00:01<00:03,   12.96it/s]
 33%|███▎      | CSD epoch blocks : 24/72 [00:01<00:03,   13.06it/s]
 35%|███▍      | CSD epoch blocks : 25/72 [00:01<00:03,   12.83it/s]
 36%|███▌      | CSD epoch blocks : 26/72 [00:02<00:03,   12.62it/s]
 38%|███▊      | CSD epoch blocks : 27/72 [00:02<00:03,   12.72it/s]
 39%|███▉      | CSD epoch blocks : 28/72 [00:02<00:03,   12.80it/s]
 40%|████      | CSD epoch blocks : 29/72 [00:02<00:03,   12.53it/s]
 42%|████▏     | CSD epoch blocks : 30/72 [00:02<00:03,   12.64it/s]
 43%|████▎     | CSD epoch blocks : 31/72 [00:02<00:03,   12.56it/s]
 44%|████▍     | CSD epoch blocks : 32/72 [00:02<00:03,   12.56it/s]
 46%|████▌     | CSD epoch blocks : 33/72 [00:02<00:03,   12.62it/s]
 47%|████▋     | CSD epoch blocks : 34/72 [00:02<00:02,   12.71it/s]
 49%|████▊     | CSD epoch blocks : 35/72 [00:02<00:02,   12.47it/s]
 50%|█████     | CSD epoch blocks : 36/72 [00:02<00:02,   12.45it/s]
 51%|█████▏    | CSD epoch blocks : 37/72 [00:02<00:02,   12.27it/s]
 53%|█████▎    | CSD epoch blocks : 38/72 [00:03<00:02,   12.37it/s]
 54%|█████▍    | CSD epoch blocks : 39/72 [00:03<00:02,   12.49it/s]
 56%|█████▌    | CSD epoch blocks : 40/72 [00:03<00:02,   12.60it/s]
 57%|█████▋    | CSD epoch blocks : 41/72 [00:03<00:02,   12.70it/s]
 58%|█████▊    | CSD epoch blocks : 42/72 [00:03<00:02,   12.77it/s]
 60%|█████▉    | CSD epoch blocks : 43/72 [00:03<00:02,   12.82it/s]
 61%|██████    | CSD epoch blocks : 44/72 [00:03<00:02,   12.87it/s]
 62%|██████▎   | CSD epoch blocks : 45/72 [00:03<00:02,   12.91it/s]
 64%|██████▍   | CSD epoch blocks : 46/72 [00:03<00:02,   12.98it/s]
 65%|██████▌   | CSD epoch blocks : 47/72 [00:03<00:01,   13.03it/s]
 67%|██████▋   | CSD epoch blocks : 48/72 [00:03<00:01,   13.10it/s]
 68%|██████▊   | CSD epoch blocks : 49/72 [00:03<00:01,   13.15it/s]
 69%|██████▉   | CSD epoch blocks : 50/72 [00:03<00:01,   13.18it/s]
 71%|███████   | CSD epoch blocks : 51/72 [00:03<00:01,   13.23it/s]
 72%|███████▏  | CSD epoch blocks : 52/72 [00:04<00:01,   13.28it/s]
 74%|███████▎  | CSD epoch blocks : 53/72 [00:04<00:01,   13.31it/s]
 75%|███████▌  | CSD epoch blocks : 54/72 [00:04<00:01,   13.26it/s]
 76%|███████▋  | CSD epoch blocks : 55/72 [00:04<00:01,   13.06it/s]
 78%|███████▊  | CSD epoch blocks : 56/72 [00:04<00:01,   13.13it/s]
 79%|███████▉  | CSD epoch blocks : 57/72 [00:04<00:01,   12.89it/s]
 81%|████████  | CSD epoch blocks : 58/72 [00:04<00:01,   12.58it/s]
 82%|████████▏ | CSD epoch blocks : 59/72 [00:04<00:01,   12.28it/s]
 83%|████████▎ | CSD epoch blocks : 60/72 [00:04<00:00,   12.25it/s]
 85%|████████▍ | CSD epoch blocks : 61/72 [00:04<00:00,   12.35it/s]
 86%|████████▌ | CSD epoch blocks : 62/72 [00:04<00:00,   12.44it/s]
 88%|████████▊ | CSD epoch blocks : 63/72 [00:04<00:00,   12.49it/s]
 89%|████████▉ | CSD epoch blocks : 64/72 [00:05<00:00,   12.57it/s]
 90%|█████████ | CSD epoch blocks : 65/72 [00:05<00:00,   12.61it/s]
 92%|█████████▏| CSD epoch blocks : 66/72 [00:05<00:00,   12.70it/s]
 93%|█████████▎| CSD epoch blocks : 67/72 [00:05<00:00,   12.76it/s]
 94%|█████████▍| CSD epoch blocks : 68/72 [00:05<00:00,   12.79it/s]
 96%|█████████▌| CSD epoch blocks : 69/72 [00:05<00:00,   12.75it/s]
 97%|█████████▋| CSD epoch blocks : 70/72 [00:05<00:00,   12.82it/s]
 99%|█████████▊| CSD epoch blocks : 71/72 [00:05<00:00,   12.89it/s]
100%|██████████| CSD epoch blocks : 72/72 [00:05<00:00,   12.84it/s]
100%|██████████| CSD epoch blocks : 72/72 [00:05<00:00,   12.81it/s]
[done]

The resulting mne.time_frequency.CrossSpectralDensity objects have a plotting function we can use to compare the results of the different methods. We’re plotting the mean CSD across frequencies. mne.time_frequency.CrossSpectralDensity.plot() returns a list of created figures; in this case, each returned list has only one figure so we use a Python trick of including a comma after our variable name to assign the figure (not the list) to our fig variable:

plot_dict = {
    "Short-time Fourier transform": csd_fft,
    "Adaptive multitapers": csd_mt,
    "Morlet wavelet transform": csd_wav,
}
for title, csd in plot_dict.items():
    (fig,) = csd.mean().plot()
    fig.suptitle(title)
  • Short-time Fourier transform, 15.8-20.0 Hz.
  • Adaptive multitapers, 15.8-20.0 Hz.
  • Morlet wavelet transform, 16.0-20.0 Hz.

Total running time of the script: (0 minutes 22.897 seconds)

Estimated memory usage: 249 MB

Gallery generated by Sphinx-Gallery