python - 从注释访问原始数据而不创建时期

Question

我想在 REM 睡眠阶段计算 EEG 信号的 PSD。我有一个带有睡眠阶段注释的原始脑电图信号。我的注释的描述是'S0，S1，S2，S3，REM'。有没有办法直接从注释中访问原始信号而不创建时期？我的意思是，如果可以根据注释的描述，直接在原始信号上计算 psd，指定我想在原始信号的哪个部分计算 PSD 。像这样的东西：

annotations = mne.Annotations(onset, duration, description, orig_time=None)
raw.set_annotations(annotations)
psds, freqs = mne.time_frequency.psd_welch(raw[raw.annotations.description['REM']], fmin=13.0, fmax=22.0)

但是上面的代码不起作用