用Python玩转DEAP情绪数据集：从数据下载到EEG信号可视化（附完整代码）

用Python玩转DEAP情绪数据集从数据下载到EEG信号可视化附完整代码在探索情感计算与脑机接口的交叉领域时DEAPDatabase for Emotion Analysis using Physiological Signals数据集无疑是一座值得深入挖掘的金矿。这个由多国研究团队联合构建的开放数据集记录了32名受试者在观看音乐视频时的多通道生理信号包括32导联EEG、眼电、皮电等丰富数据维度。对于刚接触生物信号分析的Python开发者而言如何快速上手这个复杂数据集往往面临三大挑战数据获取的迷宫、二进制文件的解析障碍以及信号可视化的技术门槛。本文将用纯代码驱动的方式带您绕过这些新手墙在Jupyter Notebook中完成从数据加载到动态频谱图生成的全流程实战。1. 数据获取与环境配置1.1 快速获取DEAP官方数据访问DEAP官网需通过学术机构邮箱注册后我们会发现数据集以两种预处理版本提供Matlab格式的.mat文件和Python优化的.dat文件。对于Python技术栈用户推荐直接下载data_preprocessed_python版本其二进制格式更适合高效读取。以下是自动下载和解压的终端操作# 创建项目目录结构 mkdir -p deap_project/{data,scripts,output} cd deap_project/data # 使用wget下载压缩包需替换为实际获取的下载链接 wget http://example.com/deap_data.zip -O deap_python.zip # 解压并验证文件结构 unzip deap_python.zip tree -L 2典型解压后的目录应包含32个.dat文件对应每位受试者和配套的元数据文件。关键文件说明文件类型描述示例文件名.dat预处理后的生理信号二进制数据s01.dat.info实验参数与标签的JSON格式元数据s01.info1.2 搭建Python分析环境推荐使用conda创建专属环境确保依赖库版本兼容# 创建并激活环境 conda create -n deap_analysis python3.8 conda activate deap_analysis # 安装核心工具链 pip install numpy pandas matplotlib mne scipy sklearn注意MNEMNE-Python是处理EEG信号的瑞士军刀其1.0版本对DEAP数据有更好的支持2. 数据加载与结构解析2.1 破解.dat二进制格式DEAP的Python预处理数据采用自定义二进制格式存储需用NumPy的memmap方法高效读取。以下代码封装了数据加载过程import numpy as np import json def load_deap_subject(data_path, subj_id): 加载单个受试者数据 # 二进制数据内存映射 data_memmap np.memmap(f{data_path}/s{subj_id:02d}.dat, dtypefloat32, moder) data data_memmap.reshape(40, 40, 8064) # 重塑为(试验×通道×采样点) # 加载元数据 with open(f{data_path}/s{subj_id:02d}.info) as f: labels json.load(f) return data, labels # 示例加载第1位受试者数据 eeg_data, metadata load_deap_subject(data_preprocessed_python, 1)数据立方体的三维结构解析第一维度40对应40段音乐视频刺激试验第二维度40生理信号通道前32个为EEG第三维度8064128Hz采样率下63秒的数据点含3秒基线2.2 通道映射与元数据提取DEAP的通道顺序遵循特定布局需要正确映射到电极位置# 标准EEG通道名称按DEAP顺序 eeg_channels [ Fp1,AF3,F3,F7,FC5,FC1,C3,T7,CP5,CP1,P3,P7,PO3, O1,Oz,Pz,Fp2,AF4,Fz,F4,F8,FC6,FC2,Cz,C4,T8, CP6,CP2,P4,P8,PO4,O2 ] # 创建通道-索引映射字典 channel_map {name: idx for idx, name in enumerate(eeg_channels)} # 提取首个试验的EEG信号所有通道 trial_idx 0 eeg_trial eeg_data[trial_idx, :32, :] # 前32通道为EEG元数据中包含关键的效价Valence、唤醒度Arousal评分可通过Pandas结构化展示import pandas as pd # 构建试验标签DataFrame trials_df pd.DataFrame({ valence: metadata[labels][:, 0], arousal: metadata[labels][:, 1], dominance: metadata[labels][:, 2], liking: metadata[labels][:, 3] }) # 查看前5个试验的标签统计 print(trials_df.head().describe())3. EEG信号预处理实战3.1 基线校正与带通滤波原始信号通常需要以下预处理步骤基线校正减去前3秒静息期的均值带通滤波保留0.5-45Hz有效频段去除直流偏移与高频噪声使用MNE库高效实现import mne from mne.filter import filter_data # 创建虚拟原始对象 sfreq 128 # 采样率 info mne.create_info(eeg_channels, sfreq, ch_typeseeg) raw mne.io.RawArray(eeg_trial, info) # 带通滤波 raw.filter(0.5, 45., fir_designfirwin) # 提取处理后的数据 filtered_data raw.get_data() # 基线校正前3秒作为基线 baseline filtered_data[:, :384].mean(axis1, keepdimsTrue) eeg_corrected filtered_data - baseline3.2 伪迹检测与处理眼电EOG和肌电EMG是EEG常见干扰源。虽然DEAP已进行预处理但仍建议进行补充检测# 计算各通道峰峰值作为伪迹指标 peak_to_peak eeg_corrected.ptp(axis1) # 标记异常通道阈值设为中位数±3倍MAD median_pp np.median(peak_to_peak) mad 1.4826 * np.median(np.abs(peak_to_peak - median_pp)) bad_channels np.where(peak_to_peak median_pp 3 * mad)[0] print(f检测到异常通道{[eeg_channels[i] for i in bad_channels]})提示对于严重污染的通道可采用插值法修复。MNE的interpolate_bads()方法可自动完成此操作4. 多维可视化技术4.1 时域信号动态绘制使用Matplotlib创建可交互的通道对比视图import matplotlib.pyplot as plt from matplotlib.widgets import Slider # 创建带滑动条的时间轴视图 fig, ax plt.subplots(figsize(12, 8)) plt.subplots_adjust(bottom0.25) # 绘制前5秒的F3通道数据 channel F3 ch_idx channel_map[channel] time_window 5 * sfreq # 5秒数据点 line, ax.plot(eeg_corrected[ch_idx, :time_window]) # 添加通道选择滑动条 axcolor lightgoldenrodyellow ax_ch plt.axes([0.25, 0.1, 0.65, 0.03], facecoloraxcolor) ch_slider Slider(ax_ch, Channel, 0, 31, valinitch_idx, valstep1) def update(val): ch int(ch_slider.val) line.set_ydata(eeg_corrected[ch, :time_window]) ax.set_ylabel(fAmplitude (μV) - {eeg_channels[ch]}) fig.canvas.draw_idle() ch_slider.on_changed(update) ax.set_xlabel(Time (samples)) plt.show()4.2 频域能量图谱通过短时傅里叶变换STFT观察不同情绪状态下的频谱特征差异from scipy import signal # 计算STFT f, t, Zxx signal.stft(eeg_corrected[channel_map[Pz]], fssfreq, nperseg256) # 绘制热力图 plt.figure(figsize(10, 6)) plt.pcolormesh(t, f, np.abs(Zxx), shadinggouraud, cmapjet) plt.colorbar(labelMagnitude (μV)) plt.ylabel(Frequency [Hz]) plt.xlabel(Time [sec]) plt.title(Pz电极时频分析) plt.ylim(0, 45) # 聚焦有效频段 plt.show()4.3 情绪维度关联分析将生理信号特征与主观评分关联挖掘潜在模式# 提取alpha波段能量作为特征 (8-13Hz) alpha_power np.mean(np.abs(Zxx)[(f 8) (f 13)], axis0) # 获取对应试验的效价评分 valence_score metadata[labels][trial_idx, 0] # 创建关联散点图 plt.scatter(alpha_power, [valence_score]*len(alpha_power), alpha0.5) plt.xlabel(Alpha波段能量 (μV^2/Hz)) plt.ylabel(效价评分 (1-9)) plt.title(EEG特征与情绪评分关联) plt.grid(True)5. 高级分析技巧5.1 跨试验数据聚合为分析特定情绪状态下的EEG模式可按评分分组数据# 定义高低效价组基于中位数分割 high_valence trials_df[trials_df.valence trials_df.valence.median()].index low_valence trials_df[trials_df.valence trials_df.valence.median()].index # 计算组间差异 def extract_band_power(data, band): f, Pxx signal.welch(data, fssfreq, nperseg256) return np.mean(Pxx[(f band[0]) (f band[1])]) # 比较两组在theta波段(4-8Hz)的差异 theta_power { high_valence: [extract_band_power(eeg_data[i,:32,:], (4,8)) for i in high_valence], low_valence: [extract_band_power(eeg_data[i,:32,:], (4,8)) for i in low_valence] } # 绘制箱线图对比 pd.DataFrame(theta_power).boxplot() plt.ylabel(Theta Power (μV^2/Hz)) plt.title(不同效价组的脑电功率差异)5.2 实时可视化管道构建可复用的数据处理管道适合Jupyter Notebook交互探索from IPython.display import display, clear_output import ipywidgets as widgets class DEAPExplorer: def __init__(self, data_path): self.data, self.labels load_deap_subject(data_path, 1) self.fig, self.ax plt.subplots(2, 1, figsize(12, 8)) # 创建交互控件 self.channel_dropdown widgets.Dropdown( optionseeg_channels, valueFz, descriptionChannel: ) self.trial_slider widgets.IntSlider( min0, max39, step1, value0, descriptionTrial: ) # 绑定事件 widgets.interactive_output( self.update_plot, {channel: self.channel_dropdown, trial: self.trial_slider} ) def update_plot(self, channel, trial): self.ax[0].clear() self.ax[1].clear() # 时域图 ch_idx channel_map[channel] self.ax[0].plot(self.data[trial, ch_idx, :]) self.ax[0].set_title(f{channel}通道时域信号 (试验{trial1})) # 频域图 f, Pxx signal.welch(self.data[trial, ch_idx, :], fssfreq) self.ax[1].semilogy(f, Pxx) self.ax[1].set_xlim(0, 45) self.ax[1].set_title(功率谱密度) clear_output(waitTrue) display(self.fig) # 启动交互界面 explorer DEAPExplorer(data_preprocessed_python) display(explorer.channel_dropdown, explorer.trial_slider)