在同一圖中為 python 用 mne 繪制多個 PSD

Question

我想使用來自 MNE python 的 plot_psd() 獲得 plot 多個 PSD。

我嘗試了以下代碼

import matplotlib.gridspec as gridspec

gs = gridspec.GridSpec(3,1)
plt.figure()
ax = plt.axes()

# First plot
ax1 = fig.add_subplot(gs[0]
raw_egi.plot_psd(ax=ax1)

ax2=fig.add_subplot(gs[1]
raw_ws_ds_hp_lp.plot_psd(ax=ax2)

ax3= fig.add_subplot(gs[2]
raw_ws_ds_hp_lp_nf.plot_psd(ax=ax3)

plt.show()

它告訴我語法無效。

以下代碼有效，但所有圖都是疊加的

import matplotlib.gridspec as gridspec

gs = gridspec.GridSpec(3,1)
plt.figure()
ax = plt.axes()

# First plot
raw_egi.plot_psd(ax=ax)

raw_ws_ds_hp_lp.plot_psd(ax=ax)

raw_ws_ds_hp_lp_nf.plot_psd(ax=ax)

plt.show()

你能告訴我 ho 到 plot 這 3 個數字沒有疊加而是垂直（一個一行）。 在下面你會找到帶有工作代碼的圖（即 3 個疊加圖）謝謝你的幫助

Answer 1

這是我如何解決 2 個地塊的問題

import matplotlib.pyplot as plt

fig, ax = plt.subplots(2)

raw_bp.plot_psd(ax=ax[0], show=False)
raw_bp_nf.plot_psd(ax=ax[1], show=False)

ax[0].set_title('PSD before filtering')
ax[1].set_title('PSD after filtering')
ax[1].set_xlabel('Frequency (Hz)')
fig.set_tight_layout(True)
plt.show()