Compute and visualize ERDS maps

This example calculates and displays ERDS maps of event-related EEG data. ERDS (sometimes also written as ERD/ERS) is short for event-related desynchronization (ERD) and event-related synchronization (ERS) 1. Conceptually, ERD corresponds to a decrease in power in a specific frequency band relative to a baseline. Similarly, ERS corresponds to an increase in power. An ERDS map is a time/frequency representation of ERD/ERS over a range of frequencies 2. ERDS maps are also known as ERSP (event-related spectral perturbation) 3.

We use a public EEG BCI data set containing two different motor imagery tasks available at PhysioNet. The two tasks are imagined hand and feet movement. Our goal is to generate ERDS maps for each of the two tasks.

First, we load the data and create epochs of 5s length. The data sets contain multiple channels, but we will only consider the three channels C3, Cz, and C4. We compute maps containing frequencies ranging from 2 to 35Hz. We map ERD to red color and ERS to blue color, which is the convention in many ERDS publications. Finally, we perform cluster-based permutation tests to estimate significant ERDS values (corrected for multiple comparisons within channels).

# Authors: Clemens Brunner <clemens.brunner@gmail.com>
#          Felix Klotzsche <klotzsche@cbs.mpg.de>
#
# License: BSD (3-clause)

import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
import seaborn as sns
import mne
from mne.datasets import eegbci
from mne.io import concatenate_raws, read_raw_edf
from mne.time_frequency import tfr_multitaper
from mne.stats import permutation_cluster_1samp_test as pcluster_test
from mne.viz.utils import center_cmap


# load and preprocess data ####################################################
subject = 1  # use data from subject 1
runs = [6, 10, 14]  # use only hand and feet motor imagery runs

fnames = eegbci.load_data(subject, runs)
raws = [read_raw_edf(f, preload=True) for f in fnames]
raw = concatenate_raws(raws)

raw.rename_channels(lambda x: x.strip('.'))  # remove dots from channel names

events, _ = mne.events_from_annotations(raw, event_id=dict(T1=2, T2=3))

want_chs = ['C3', 'Cz', 'C4']
picks = mne.pick_channels(raw.info["ch_names"], want_chs)

# epoch data ##################################################################
tmin, tmax = -1, 4  # define epochs around events (in s)
event_ids = dict(hands=2, feet=3)  # map event IDs to tasks

epochs = mne.Epochs(raw, events, event_ids, tmin - 0.5, tmax + 0.5,
                    picks=picks, baseline=None, preload=True)

# compute ERDS maps ###########################################################
freqs = np.arange(2, 36, 1)  # frequencies from 2-35Hz
n_cycles = freqs  # use constant t/f resolution
vmin, vmax = -1, 1.5  # set min and max ERDS values in plot
baseline = [-1, 0]  # baseline interval (in s)
cmap = center_cmap(plt.cm.RdBu, vmin, vmax)  # zero maps to white
kwargs = dict(n_permutations=100, step_down_p=0.05, seed=1,
              buffer_size=None, out_type='mask')  # for cluster test

# Run TF decomposition overall epochs
tfr = tfr_multitaper(epochs, freqs=freqs, n_cycles=n_cycles,
                     use_fft=True, return_itc=False, average=False,
                     decim=2)
tfr.crop(tmin, tmax)
tfr.apply_baseline(baseline, mode="percent")
for event in event_ids:
    # select desired epochs for visualization
    tfr_ev = tfr[event]
    fig, axes = plt.subplots(1, 4, figsize=(12, 4),
                             gridspec_kw={"width_ratios": [10, 10, 10, 1]})
    for ch, ax in enumerate(axes[:-1]):  # for each channel
        # positive clusters
        _, c1, p1, _ = pcluster_test(tfr_ev.data[:, ch, ...], tail=1, **kwargs)
        # negative clusters
        _, c2, p2, _ = pcluster_test(tfr_ev.data[:, ch, ...], tail=-1,
                                     **kwargs)

        # note that we keep clusters with p <= 0.05 from the combined clusters
        # of two independent tests; in this example, we do not correct for
        # these two comparisons
        c = np.stack(c1 + c2, axis=2)  # combined clusters
        p = np.concatenate((p1, p2))  # combined p-values
        mask = c[..., p <= 0.05].any(axis=-1)

        # plot TFR (ERDS map with masking)
        tfr_ev.average().plot([ch], vmin=vmin, vmax=vmax, cmap=(cmap, False),
                              axes=ax, colorbar=False, show=False, mask=mask,
                              mask_style="mask")

        ax.set_title(epochs.ch_names[ch], fontsize=10)
        ax.axvline(0, linewidth=1, color="black", linestyle=":")  # event
        if ch != 0:
            ax.set_ylabel("")
            ax.set_yticklabels("")
    fig.colorbar(axes[0].images[-1], cax=axes[-1])
    fig.suptitle("ERDS ({})".format(event))
    fig.show()
  • ERDS (hands), C3, Cz, C4
  • ERDS (feet), C3, Cz, C4

Out:

Extracting EDF parameters from /home/circleci/mne_data/MNE-eegbci-data/files/eegmmidb/1.0.0/S001/S001R06.edf...
EDF file detected
Setting channel info structure...
Creating raw.info structure...
Reading 0 ... 19999  =      0.000 ...   124.994 secs...
Extracting EDF parameters from /home/circleci/mne_data/MNE-eegbci-data/files/eegmmidb/1.0.0/S001/S001R10.edf...
EDF file detected
Setting channel info structure...
Creating raw.info structure...
Reading 0 ... 19999  =      0.000 ...   124.994 secs...
Extracting EDF parameters from /home/circleci/mne_data/MNE-eegbci-data/files/eegmmidb/1.0.0/S001/S001R14.edf...
EDF file detected
Setting channel info structure...
Creating raw.info structure...
Reading 0 ... 19999  =      0.000 ...   124.994 secs...
Used Annotations descriptions: ['T1', 'T2']
Not setting metadata
Not setting metadata
45 matching events found
No baseline correction applied
0 projection items activated
Loading data for 45 events and 961 original time points ...
0 bad epochs dropped
Not setting metadata
Applying baseline correction (mode: percent)
Using a threshold of 1.724718
stat_fun(H1): min=-8.523637 max=3.197747
Running initial clustering
Found 78 clusters
Permuting 99 times...

  0%|          |  : 0/99 [00:00<?,       ?it/s]
  7%|7         |  : 7/99 [00:00<00:00,  201.61it/s]
 17%|#7        |  : 17/99 [00:00<00:00,  248.88it/s]
 28%|##8       |  : 28/99 [00:00<00:00,  275.43it/s]
 41%|####1     |  : 41/99 [00:00<00:00,  304.62it/s]
 55%|#####4    |  : 54/99 [00:00<00:00,  321.14it/s]
 67%|######6   |  : 66/99 [00:00<00:00,  327.43it/s]
 80%|#######9  |  : 79/99 [00:00<00:00,  336.67it/s]
 93%|#########2|  : 92/99 [00:00<00:00,  343.73it/s]
100%|##########|  : 99/99 [00:00<00:00,  347.44it/s]
100%|##########|  : 99/99 [00:00<00:00,  340.85it/s]
Computing cluster p-values
Step-down-in-jumps iteration #1 found 0 clusters to exclude from subsequent iterations
Done.
Using a threshold of -1.724718
stat_fun(H1): min=-8.523637 max=3.197747
Running initial clustering
Found 65 clusters
Permuting 99 times...

  0%|          |  : 0/99 [00:00<?,       ?it/s]
 12%|#2        |  : 12/99 [00:00<00:00,  354.56it/s]
 25%|##5       |  : 25/99 [00:00<00:00,  369.34it/s]
 36%|###6      |  : 36/99 [00:00<00:00,  353.86it/s]
 49%|####9     |  : 49/99 [00:00<00:00,  361.67it/s]
 61%|######    |  : 60/99 [00:00<00:00,  353.36it/s]
 73%|#######2  |  : 72/99 [00:00<00:00,  353.67it/s]
 85%|########4 |  : 84/99 [00:00<00:00,  353.81it/s]
 98%|#########7|  : 97/99 [00:00<00:00,  358.39it/s]
100%|##########|  : 99/99 [00:00<00:00,  357.41it/s]
Computing cluster p-values
Step-down-in-jumps iteration #1 found 1 cluster to exclude from subsequent iterations
Permuting 99 times...

  0%|          |  : 0/99 [00:00<?,       ?it/s]
 12%|#2        |  : 12/99 [00:00<00:00,  354.40it/s]
 21%|##1       |  : 21/99 [00:00<00:00,  309.20it/s]
 33%|###3      |  : 33/99 [00:00<00:00,  325.10it/s]
 41%|####1     |  : 41/99 [00:00<00:00,  301.14it/s]
 49%|####9     |  : 49/99 [00:00<00:00,  286.85it/s]
 57%|#####6    |  : 56/99 [00:00<00:00,  271.72it/s]
 65%|######4   |  : 64/99 [00:00<00:00,  265.95it/s]
 74%|#######3  |  : 73/99 [00:00<00:00,  265.96it/s]
 87%|########6 |  : 86/99 [00:00<00:00,  281.97it/s]
 98%|#########7|  : 97/99 [00:00<00:00,  287.34it/s]
100%|##########|  : 99/99 [00:00<00:00,  289.10it/s]
Computing cluster p-values
Step-down-in-jumps iteration #2 found 0 additional clusters to exclude from subsequent iterations
Done.
No baseline correction applied
Using a threshold of 1.724718
stat_fun(H1): min=-4.573067 max=3.687727
Running initial clustering
Found 85 clusters
Permuting 99 times...

  0%|          |  : 0/99 [00:00<?,       ?it/s]
 11%|#1        |  : 11/99 [00:00<00:00,  323.94it/s]
 17%|#7        |  : 17/99 [00:00<00:00,  249.00it/s]
 30%|###       |  : 30/99 [00:00<00:00,  296.29it/s]
 42%|####2     |  : 42/99 [00:00<00:00,  312.05it/s]
 56%|#####5    |  : 55/99 [00:00<00:00,  327.94it/s]
 68%|######7   |  : 67/99 [00:00<00:00,  333.04it/s]
 80%|#######9  |  : 79/99 [00:00<00:00,  336.50it/s]
 93%|#########2|  : 92/99 [00:00<00:00,  343.60it/s]
100%|##########|  : 99/99 [00:00<00:00,  349.53it/s]
100%|##########|  : 99/99 [00:00<00:00,  344.24it/s]
Computing cluster p-values
Step-down-in-jumps iteration #1 found 1 cluster to exclude from subsequent iterations
Permuting 99 times...

  0%|          |  : 0/99 [00:00<?,       ?it/s]
 12%|#2        |  : 12/99 [00:00<00:00,  353.40it/s]
 24%|##4       |  : 24/99 [00:00<00:00,  353.53it/s]
 36%|###6      |  : 36/99 [00:00<00:00,  353.52it/s]
 49%|####9     |  : 49/99 [00:00<00:00,  361.53it/s]
 63%|######2   |  : 62/99 [00:00<00:00,  366.34it/s]
 76%|#######5  |  : 75/99 [00:00<00:00,  369.66it/s]
 88%|########7 |  : 87/99 [00:00<00:00,  367.23it/s]
100%|##########|  : 99/99 [00:00<00:00,  372.17it/s]
100%|##########|  : 99/99 [00:00<00:00,  370.24it/s]
Computing cluster p-values
Step-down-in-jumps iteration #2 found 0 additional clusters to exclude from subsequent iterations
Done.
Using a threshold of -1.724718
stat_fun(H1): min=-4.573067 max=3.687727
Running initial clustering
Found 57 clusters
Permuting 99 times...

  0%|          |  : 0/99 [00:00<?,       ?it/s]
 12%|#2        |  : 12/99 [00:00<00:00,  349.74it/s]
 25%|##5       |  : 25/99 [00:00<00:00,  367.51it/s]
 36%|###6      |  : 36/99 [00:00<00:00,  352.90it/s]
 49%|####9     |  : 49/99 [00:00<00:00,  361.39it/s]
 62%|######1   |  : 61/99 [00:00<00:00,  360.05it/s]
 75%|#######4  |  : 74/99 [00:00<00:00,  364.66it/s]
 87%|########6 |  : 86/99 [00:00<00:00,  362.86it/s]
 98%|#########7|  : 97/99 [00:00<00:00,  357.28it/s]
100%|##########|  : 99/99 [00:00<00:00,  360.22it/s]
Computing cluster p-values
Step-down-in-jumps iteration #1 found 0 clusters to exclude from subsequent iterations
Done.
No baseline correction applied
Using a threshold of 1.724718
stat_fun(H1): min=-6.599131 max=3.329547
Running initial clustering
Found 64 clusters
Permuting 99 times...

  0%|          |  : 0/99 [00:00<?,       ?it/s]
 11%|#1        |  : 11/99 [00:00<00:00,  324.25it/s]
 23%|##3       |  : 23/99 [00:00<00:00,  339.28it/s]
 35%|###5      |  : 35/99 [00:00<00:00,  344.26it/s]
 48%|####8     |  : 48/99 [00:00<00:00,  354.77it/s]
 62%|######1   |  : 61/99 [00:00<00:00,  361.09it/s]
 75%|#######4  |  : 74/99 [00:00<00:00,  361.69it/s]
 88%|########7 |  : 87/99 [00:00<00:00,  365.47it/s]
100%|##########|  : 99/99 [00:00<00:00,  366.13it/s]
100%|##########|  : 99/99 [00:00<00:00,  364.05it/s]
Computing cluster p-values
Step-down-in-jumps iteration #1 found 0 clusters to exclude from subsequent iterations
Done.
Using a threshold of -1.724718
stat_fun(H1): min=-6.599131 max=3.329547
Running initial clustering
Found 64 clusters
Permuting 99 times...

  0%|          |  : 0/99 [00:00<?,       ?it/s]
 12%|#2        |  : 12/99 [00:00<00:00,  353.24it/s]
 24%|##4       |  : 24/99 [00:00<00:00,  353.45it/s]
 36%|###6      |  : 36/99 [00:00<00:00,  353.71it/s]
 49%|####9     |  : 49/99 [00:00<00:00,  362.09it/s]
 62%|######1   |  : 61/99 [00:00<00:00,  360.43it/s]
 74%|#######3  |  : 73/99 [00:00<00:00,  359.41it/s]
 87%|########6 |  : 86/99 [00:00<00:00,  363.44it/s]
100%|##########|  : 99/99 [00:00<00:00,  368.54it/s]
100%|##########|  : 99/99 [00:00<00:00,  366.59it/s]
Computing cluster p-values
Step-down-in-jumps iteration #1 found 0 clusters to exclude from subsequent iterations
Done.
No baseline correction applied
Using a threshold of 1.713872
stat_fun(H1): min=-3.687815 max=3.369164
Running initial clustering
Found 66 clusters
Permuting 99 times...

  0%|          |  : 0/99 [00:00<?,       ?it/s]
 11%|#1        |  : 11/99 [00:00<00:00,  324.15it/s]
 23%|##3       |  : 23/99 [00:00<00:00,  340.03it/s]
 33%|###3      |  : 33/99 [00:00<00:00,  324.57it/s]
 44%|####4     |  : 44/99 [00:00<00:00,  324.78it/s]
 57%|#####6    |  : 56/99 [00:00<00:00,  331.28it/s]
 68%|######7   |  : 67/99 [00:00<00:00,  329.93it/s]
 79%|#######8  |  : 78/99 [00:00<00:00,  329.18it/s]
 91%|######### |  : 90/99 [00:00<00:00,  332.82it/s]
100%|##########|  : 99/99 [00:00<00:00,  333.01it/s]
100%|##########|  : 99/99 [00:00<00:00,  332.02it/s]
Computing cluster p-values
Step-down-in-jumps iteration #1 found 0 clusters to exclude from subsequent iterations
Done.
Using a threshold of -1.713872
stat_fun(H1): min=-3.687815 max=3.369164
Running initial clustering
Found 75 clusters
Permuting 99 times...

  0%|          |  : 0/99 [00:00<?,       ?it/s]
 10%|#         |  : 10/99 [00:00<00:00,  295.18it/s]
 21%|##1       |  : 21/99 [00:00<00:00,  310.38it/s]
 34%|###4      |  : 34/99 [00:00<00:00,  336.37it/s]
 45%|####5     |  : 45/99 [00:00<00:00,  333.43it/s]
 55%|#####4    |  : 54/99 [00:00<00:00,  318.60it/s]
 67%|######6   |  : 66/99 [00:00<00:00,  325.49it/s]
 79%|#######8  |  : 78/99 [00:00<00:00,  330.42it/s]
 91%|######### |  : 90/99 [00:00<00:00,  334.08it/s]
100%|##########|  : 99/99 [00:00<00:00,  334.95it/s]
100%|##########|  : 99/99 [00:00<00:00,  333.11it/s]
Computing cluster p-values
Step-down-in-jumps iteration #1 found 0 clusters to exclude from subsequent iterations
Done.
No baseline correction applied
Using a threshold of 1.713872
stat_fun(H1): min=-5.046259 max=5.406477
Running initial clustering
Found 98 clusters
Permuting 99 times...

  0%|          |  : 0/99 [00:00<?,       ?it/s]
 12%|#2        |  : 12/99 [00:00<00:00,  353.90it/s]
 24%|##4       |  : 24/99 [00:00<00:00,  354.49it/s]
 31%|###1      |  : 31/99 [00:00<00:00,  302.76it/s]
 38%|###8      |  : 38/99 [00:00<00:00,  276.88it/s]
 49%|####9     |  : 49/99 [00:00<00:00,  287.61it/s]
 63%|######2   |  : 62/99 [00:00<00:00,  305.92it/s]
 76%|#######5  |  : 75/99 [00:00<00:00,  318.84it/s]
 88%|########7 |  : 87/99 [00:00<00:00,  324.06it/s]
100%|##########|  : 99/99 [00:00<00:00,  332.50it/s]
100%|##########|  : 99/99 [00:00<00:00,  328.10it/s]
Computing cluster p-values
Step-down-in-jumps iteration #1 found 1 cluster to exclude from subsequent iterations
Permuting 99 times...

  0%|          |  : 0/99 [00:00<?,       ?it/s]
 12%|#2        |  : 12/99 [00:00<00:00,  354.80it/s]
 23%|##3       |  : 23/99 [00:00<00:00,  339.70it/s]
 36%|###6      |  : 36/99 [00:00<00:00,  355.33it/s]
 48%|####8     |  : 48/99 [00:00<00:00,  355.18it/s]
 61%|######    |  : 60/99 [00:00<00:00,  355.19it/s]
 74%|#######3  |  : 73/99 [00:00<00:00,  360.80it/s]
 86%|########5 |  : 85/99 [00:00<00:00,  359.81it/s]
 98%|#########7|  : 97/99 [00:00<00:00,  359.11it/s]
100%|##########|  : 99/99 [00:00<00:00,  359.66it/s]
Computing cluster p-values
Step-down-in-jumps iteration #2 found 0 additional clusters to exclude from subsequent iterations
Done.
Using a threshold of -1.713872
stat_fun(H1): min=-5.046259 max=5.406477
Running initial clustering
Found 66 clusters
Permuting 99 times...

  0%|          |  : 0/99 [00:00<?,       ?it/s]
 10%|#         |  : 10/99 [00:00<00:00,  295.25it/s]
 21%|##1       |  : 21/99 [00:00<00:00,  310.82it/s]
 34%|###4      |  : 34/99 [00:00<00:00,  336.61it/s]
 45%|####5     |  : 45/99 [00:00<00:00,  333.57it/s]
 58%|#####7    |  : 57/99 [00:00<00:00,  338.32it/s]
 70%|######9   |  : 69/99 [00:00<00:00,  341.53it/s]
 81%|########  |  : 80/99 [00:00<00:00,  338.83it/s]
 92%|#########1|  : 91/99 [00:00<00:00,  336.72it/s]
100%|##########|  : 99/99 [00:00<00:00,  337.27it/s]
100%|##########|  : 99/99 [00:00<00:00,  336.40it/s]
Computing cluster p-values
Step-down-in-jumps iteration #1 found 0 clusters to exclude from subsequent iterations
Done.
No baseline correction applied
Using a threshold of 1.713872
stat_fun(H1): min=-5.964817 max=4.078953
Running initial clustering
Found 92 clusters
Permuting 99 times...

  0%|          |  : 0/99 [00:00<?,       ?it/s]
 10%|#         |  : 10/99 [00:00<00:00,  295.18it/s]
 19%|#9        |  : 19/99 [00:00<00:00,  280.29it/s]
 30%|###       |  : 30/99 [00:00<00:00,  296.14it/s]
 42%|####2     |  : 42/99 [00:00<00:00,  311.87it/s]
 53%|#####2    |  : 52/99 [00:00<00:00,  308.22it/s]
 64%|######3   |  : 63/99 [00:00<00:00,  311.48it/s]
 76%|#######5  |  : 75/99 [00:00<00:00,  318.68it/s]
 86%|########5 |  : 85/99 [00:00<00:00,  315.31it/s]
 98%|#########7|  : 97/99 [00:00<00:00,  320.70it/s]
100%|##########|  : 99/99 [00:00<00:00,  320.61it/s]
Computing cluster p-values
Step-down-in-jumps iteration #1 found 1 cluster to exclude from subsequent iterations
Permuting 99 times...

  0%|          |  : 0/99 [00:00<?,       ?it/s]
  9%|9         |  : 9/99 [00:00<00:00,  265.81it/s]
 20%|##        |  : 20/99 [00:00<00:00,  295.84it/s]
 31%|###1      |  : 31/99 [00:00<00:00,  306.10it/s]
 40%|####      |  : 40/99 [00:00<00:00,  295.36it/s]
 52%|#####1    |  : 51/99 [00:00<00:00,  302.00it/s]
 64%|######3   |  : 63/99 [00:00<00:00,  312.01it/s]
 74%|#######3  |  : 73/99 [00:00<00:00,  309.26it/s]
 86%|########5 |  : 85/99 [00:00<00:00,  316.03it/s]
 97%|#########6|  : 96/99 [00:00<00:00,  317.24it/s]
100%|##########|  : 99/99 [00:00<00:00,  317.31it/s]
Computing cluster p-values
Step-down-in-jumps iteration #2 found 0 additional clusters to exclude from subsequent iterations
Done.
Using a threshold of -1.713872
stat_fun(H1): min=-5.964817 max=4.078953
Running initial clustering
Found 52 clusters
Permuting 99 times...

  0%|          |  : 0/99 [00:00<?,       ?it/s]
 11%|#1        |  : 11/99 [00:00<00:00,  324.33it/s]
 22%|##2       |  : 22/99 [00:00<00:00,  324.12it/s]
 34%|###4      |  : 34/99 [00:00<00:00,  334.50it/s]
 46%|####6     |  : 46/99 [00:00<00:00,  339.91it/s]
 59%|#####8    |  : 58/99 [00:00<00:00,  343.23it/s]
 69%|######8   |  : 68/99 [00:00<00:00,  334.27it/s]
 80%|#######9  |  : 79/99 [00:00<00:00,  332.82it/s]
 91%|######### |  : 90/99 [00:00<00:00,  331.69it/s]
100%|##########|  : 99/99 [00:00<00:00,  327.48it/s]
100%|##########|  : 99/99 [00:00<00:00,  328.33it/s]
Computing cluster p-values
Step-down-in-jumps iteration #1 found 0 clusters to exclude from subsequent iterations
Done.
No baseline correction applied

Similar to Epochs objects, we can also export data from EpochsTFR and AverageTFR objects to a Pandas DataFrame. By default, the time column of the exported data frame is in milliseconds. Here, to be consistent with the time-frequency plots, we want to keep it in seconds, which we can achieve by setting time_format=None:

df = tfr.to_data_frame(time_format=None)
df.head()
time freq epoch condition C3 Cz C4
0 -1.0000 2.0 0 feet -0.481570 -0.479232 -0.765757
1 -0.9875 2.0 0 feet -0.500555 -0.485289 -0.764209
2 -0.9750 2.0 0 feet -0.509981 -0.474900 -0.751235
3 -0.9625 2.0 0 feet -0.509883 -0.455039 -0.730498
4 -0.9500 2.0 0 feet -0.524450 -0.448783 -0.721221


This allows us to use additional plotting functions like seaborn.lineplot() to plot confidence bands:

df = tfr.to_data_frame(time_format=None, long_format=True)

# Map to frequency bands:
freq_bounds = {'_': 0,
               'delta': 3,
               'theta': 7,
               'alpha': 13,
               'beta': 35,
               'gamma': 140}
df['band'] = pd.cut(df['freq'], list(freq_bounds.values()),
                    labels=list(freq_bounds)[1:])

# Filter to retain only relevant frequency bands:
freq_bands_of_interest = ['delta', 'theta', 'alpha', 'beta']
df = df[df.band.isin(freq_bands_of_interest)]
df['band'] = df['band'].cat.remove_unused_categories()

# Order channels for plotting:
df['channel'] = df['channel'].cat.reorder_categories(want_chs, ordered=True)

g = sns.FacetGrid(df, row='band', col='channel', margin_titles=True)
g.map(sns.lineplot, 'time', 'value', 'condition', n_boot=10)
axline_kw = dict(color='black', linestyle='dashed', linewidth=0.5, alpha=0.5)
g.map(plt.axhline, y=0, **axline_kw)
g.map(plt.axvline, x=0, **axline_kw)
g.set(ylim=(None, 1.5))
g.set_axis_labels("Time (s)", "ERDS (%)")
g.set_titles(col_template="{col_name}", row_template="{row_name}")
g.add_legend(ncol=2, loc='lower center')
g.fig.subplots_adjust(left=0.1, right=0.9, top=0.9, bottom=0.08)
C3, Cz, C4

Out:

Converting "condition" to "category"...
Converting "epoch" to "category"...
Converting "freq" to "category"...
Converting "channel" to "category"...
Converting "ch_type" to "category"...

Having the data as a DataFrame also facilitates subsetting, grouping, and other transforms. Here, we use seaborn to plot the average ERDS in the motor imagery interval as a function of frequency band and imagery condition:

df_mean = (df.query('time > 1')
             .groupby(['condition', 'epoch', 'band', 'channel'])[['value']]
             .mean()
             .reset_index())

g = sns.FacetGrid(df_mean, col='condition', col_order=['hands', 'feet'],
                  margin_titles=True)
g = (g.map(sns.violinplot, 'channel', 'value', 'band', n_boot=10,
           palette='deep', order=['C3', 'Cz', 'C4'],
           hue_order=freq_bands_of_interest,
           linewidth=0.5)
      .add_legend(ncol=4, loc='lower center'))

g.map(plt.axhline, **axline_kw)
g.set_axis_labels("", "ERDS (%)")
g.set_titles(col_template="{col_name}", row_template="{row_name}")
g.fig.subplots_adjust(left=0.1, right=0.9, top=0.9, bottom=0.3)
hands, feet

References

1

Gert Pfurtscheller and Fernando H. Lopes da Silva. Event-related EEG/MEG synchronization and desynchronization: basic principles. Clinical Neurophysiology, 110(11):1842–1857, 1999. doi:10.1016/S1388-2457(99)00141-8.

2

Bernhard Graimann, Jane E. Huggins, Simon P. Levine, and Gert Pfurtscheller. Visualization of significant ERD/ERS patterns in multichannel EEG and ECoG data. Clinical Neurophysiology, 113(1):43–47, 2002. doi:10.1016/S1388-2457(01)00697-6.

3

Scott Makeig. Auditory event-related dynamics of the EEG spectrum and effects of exposure to tones. Electroencephalography and Clinical Neurophysiology, 86(4):283–293, 1993. doi:10.1016/0013-4694(93)90110-H.

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

Estimated memory usage: 120 MB

Gallery generated by Sphinx-Gallery