做了個(gè)腦地形圖，里面涉及到了一些關(guān)于matplotlib子圖繪制、colorbar設(shè)置以及自定義colormap等問(wèn)題，列在這里了先

文章目錄

• • 讀取.mat文件
  • 拓?fù)鋱D顯示函數(shù)
  • 顯示每個(gè)頻率腦地形圖
  • 各個(gè)頻帶平均功率
  • 前后相減平均值
  • 配對(duì)檢驗(yàn)并顯示
  • • P值
    • T值
  • 每一部分的圖
  • • 每個(gè)人各個(gè)頻帶
    • 各頻帶平均
    • 前后相減平均值
    • 配對(duì)檢驗(yàn)-p值
    • 配對(duì)檢驗(yàn)-t值

讀取.mat文件

import scipy.io as sio # 這里根據(jù)具體的結(jié)構(gòu)體結(jié)構(gòu)來(lái)分析 bp_result = sio.loadmat(r"bp_result.mat")["result"][0][0].tolist() bp_result = np.array(bp_result).reshape(11) events_name = ["EC_before" , "EO_before" , "EC_after" , "EO_after"] band_name = ["δ" , "θ" , "α" , "β" , "γ"] # result 索引層級(jí)： event 人 通道 頻帶 # 歸一化 for event in events_name:for i in range(len(bp_result[event])):bp_result[event][i] /= np.sum(bp_result[event][i] , axis = 1).reshape(32 , 1)

拓?fù)鋱D顯示函數(shù)

p.s. 這里用了mne庫(kù)

根據(jù)我自己的需要選擇了一些輸入?yún)?shù)。

def show_topo(data , vmin = 0 , vmax = 1 , contours = 6 , cmap = 'jet'):# plt.figure(figsize = (10 , 10))# 后面的操作不接受一維變量data = data.reshape(32 , 1)# 創(chuàng)建信息info = mne.create_info(ch_names = ch_names , sfreq = 1000 , ch_types = 'eeg')# 創(chuàng)建evoke對(duì)象evoked = mne.EvokedArray(data , info)# 設(shè)置電極evoked.set_montage(biosemi_montage)# 顯示拓?fù)鋱Dax , countour = mne.viz.plot_topomap(evoked.data[: , 0] ,evoked.info , show = False , cmap = cmap , vmin = vmin , vmax = vmax , contours = contours) return ax , countour

顯示每個(gè)頻率腦地形圖

# 支持中文 plt.rcParams['font.sans-serif'] = ['SimHei'] # 用來(lái)正常顯示中文標(biāo)簽 plt.rcParams['axes.unicode_minus'] = False # 用來(lái)正常顯示負(fù)號(hào)for event in events_name:fig = plt.figure(figsize = (10 , 20) , dpi = 300)for j in range(11):for i in range(5):data = bp_result[event][j][: , i]ax = plt.subplot(11 , 5 , j * 5 + i + 1 )aximage , countour = show_topo(data , vmin = 0 , vmax = 0.5)if j == 0:ax.set_title(band_name[i] , fontsize = 15)#colorbar 左 下 寬 高 l = 0.92b = 0.2w = 0.015h = 0.6#對(duì)應(yīng) l,b,w,h；設(shè)置colorbar位置；rect = [l,b,w,h] cbar_ax = fig.add_axes(rect) plt.colorbar(aximage, cax=cbar_ax)fig.suptitle(event + " 每個(gè)人各個(gè)頻帶腦地形圖" , fontsize="x-large")plt.savefig("result/" + event + "_每個(gè)人各個(gè)頻帶腦地形圖.png")

各個(gè)頻帶平均功率

fig = plt.figure(figsize = (10 , 7) , dpi = 300) for j , event in enumerate(events_name):for i in range(5):data = np.average(bp_result[event])[: , i]ax = plt.subplot(4 , 5 ,j * 5 + i + 1 )aximage , countour = show_topo(data , vmin = 0 , vmax = 0.4)if j == 1:ax.set_title(band_name[i] , fontsize = 15)if i == 0:ax.set_ylabel(event , rotation = 90 , size = 'large') #colorbar 左 下 寬 高 l = 0.92 b = 0.2 w = 0.015 h = 0.6 #對(duì)應(yīng) l,b,w,h；設(shè)置colorbar位置； rect = [l,b,w,h] cbar_ax = fig.add_axes(rect) plt.colorbar(aximage, cax=cbar_ax) fig.suptitle("各頻帶平均" , fontsize="x-large") plt.savefig("result/" + "各個(gè)頻帶腦地形圖_平均.png")

前后相減平均值

fig = plt.figure(figsize = (10 , 5) , dpi = 300) for j in [0 , 1]:for i in range(5):# 計(jì)算前后差值data = np.average(bp_result[events_name[j + 2]] - bp_result[events_name[j]])[: , i]ax = plt.subplot(2 , 5 ,j * 5 + i + 1 )aximage , countour = show_topo(data , vmin = -0.15, vmax = 0.15)if j == 0:ax.set_title(band_name[i] , fontsize = 15)if i == 0:ax.set_ylabel("%s"%(events_name[j + 2][:2]) , rotation = 90 , size = 'large') #colorbar 左 下 寬 高 l = 0.92 b = 0.2 w = 0.015 h = 0.6 #對(duì)應(yīng) l,b,w,h；設(shè)置colorbar位置； rect = [l,b,w,h] cbar_ax = fig.add_axes(rect) plt.colorbar(aximage, cax=cbar_ax) fig.suptitle("after - before" , fontsize="x-large") plt.savefig("result/" + "各個(gè)頻帶腦地形圖_平均差異.png")

配對(duì)檢驗(yàn)并顯示

P值

import matplotlib as mpl fig = plt.figure(figsize = (10 , 5) , dpi = 300) for j in [0 , 1]:d = np.zeros((11 , 32 , 5))temp = bp_result[events_name[j + 2]] - bp_result[events_name[j]]for i in range(len(bp_result[events_name[j]])):d[i , : , :] = temp[i]for i in range(5):# 計(jì)算前后差值t , p_two = stats.ttest_1samp(d[: , : , i] , 0)data = p_twoax = plt.subplot(2 , 5 ,j * 5 + i + 1 )# 自行創(chuàng)建colormap，第一個(gè)參數(shù)是16進(jìn)制RGB顏色， 第二個(gè)參數(shù)是分成幾個(gè)顏色塊C = mpl.colors.LinearSegmentedColormap.from_list('cmap', ['#FF0000', '#FFFF00', '#00F5FF', '#FFFFFF'], 6)aximage , countour = show_topo(data , vmin = 0, vmax = 0.06 ,cmap = C)if j == 0:ax.set_title(band_name[i] , fontsize = 15)if i == 0:ax.set_ylabel("%s"%(events_name[j + 2][:2]) , rotation = 90 , size = 'large') #colorbar 左 下 寬 高 l = 0.92 b = 0.2 w = 0.015 h = 0.6 #對(duì)應(yīng) l,b,w,h；設(shè)置colorbar位置； rect = [l,b,w,h] cbar_ax = fig.add_axes(rect) plt.colorbar(aximage, cax=cbar_ax) fig.suptitle("after - before p_val" , fontsize="x-large") plt.savefig("result/" + "各個(gè)頻帶腦地形圖_平均差異_p.png")

T值

fig = plt.figure(figsize = (10 , 5) , dpi = 300) for j in [0 , 1]:d = np.zeros((11 , 32 , 5))temp = bp_result[events_name[j + 2]] - bp_result[events_name[j]]for i in range(len(bp_result[events_name[j]])):d[i , : , :] = temp[i]for i in range(5):# 計(jì)算前后差值t , p_two = stats.ttest_1samp(d[: , : , i] , 0)data = tax = plt.subplot(2 , 5 ,j * 5 + i + 1 )aximage , countour = show_topo(data , vmin = -3 , vmax = 3 ,cmap = 'bwr')if j == 0:ax.set_title(band_name[i] , fontsize = 15)if i == 0:ax.set_ylabel("%s"%(events_name[j + 2][:2]) , rotation = 90 , size = 'large') #colorbar 左 下 寬 高 l = 0.92 b = 0.2 w = 0.015 h = 0.6 #對(duì)應(yīng) l,b,w,h；設(shè)置colorbar位置； rect = [l,b,w,h] cbar_ax = fig.add_axes(rect) plt.colorbar(aximage, cax=cbar_ax) fig.suptitle("after - before t_val" , fontsize="x-large") plt.savefig("result/" + "各個(gè)頻帶腦地形圖_平均差異_t.png")

每一部分的圖

每個(gè)人各個(gè)頻帶

各頻帶平均

前后相減平均值

配對(duì)檢驗(yàn)-p值

配對(duì)檢驗(yàn)-t值

總結(jié)

以上是生活随笔為你收集整理的基于python的脑电地形图显示的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。