python制作脑电相关图
过程如下
- 素材:
(1) 清华数据库Benchmark下文件:64-channels.loc
(2) 一个trial的64*1500数据 - 代码:
import mne
import matplotlib.pyplot as plt
import scipy.io
path = r'data.mat'
raw_data = scipy.io.loadmat(path)
montage=mne.channels.read_custom_montage("64-channels.loc")
## 制作info是关键灵魂,需要data和info
info = mne.create_info(
ch_names=montage.ch_names,
ch_types=['eeg']*64,
sfreq=250
)
data =raw_data['data']
raw = mne.io.RawArray(data, info)
raw.set_montage(montage)
print(raw.info['ch_names'])
raw.plot_sensors(show_names=True)
# layout_from_raw = mne.channels.make_eeg_layout(raw.info)
# layout_from_raw.plot()
plt.show()
- 效果图
- 打印psd
raw.plot_psd()
# raw.plot_psd(fmin=1, fmax=90, tmin=0.5,tmax=6.0,spatial_colors=True)
plt.show()