代码
import mne
import numpy as np
import scipy.io
%pylab inline
data = scipy.io.loadmat('dataset_BCIcomp1.mat')
np.samples = data['x_train']
ch_names = ['C3','C4','CZ']
sfreq = 128
info = mne.create_info(ch_names, sfreq)
for k in range(140):
np.C3 = [0 for _ in range(1152)]
np.C4 = [0 for _ in range(1152)]
np.CZ = [0 for _ in range(1152)]
np.C = []
for i,t in enumerate(np.samples):
np.C3[i] = np.samples[i][0][k]
np.C4[i] = np.samples[i][1][k]
np.CZ[i] = np.samples[i][2][k]
np.C = np.vstack((np.C3,np.C4))
np.C = np.vstack((np.C,np.CZ))
raw = mne.io.RawArray(np.C, info)
raw.plot()
print('数据集的形状为:',raw.get_data().shape)
print('通道数为:',raw.info.get('nchan'))
一个被试的C3、C4、CZ三通道脑电数据图