可视化代码

可视化.py

@@ -0,0 +1,26 @@
+import numpy as np
+import mne
+import matplotlib.pyplot as plt
+# loading data
+
+
+data = np.load("ABMD_results.npy", allow_pickle=True).item()
+samples = data['data']
+
+cnt = 1
+
+for sample in samples:
+    # 创建一个列表，其中包含30个eeg1到eeg30的名称
+    channels = [f"eeg{i}" for i in range(1, 31)]
+    # 创建一个列表，其中包含30个eeg类型的元素
+    channel_types = ["eeg"] * 30
+    # 使用mne模块创建一个info对象，其中包含上面创建的channels和channel_types列表
+    info = mne.create_info(ch_names=channels,
+                           sfreq=1100,
+                           ch_types=channel_types)
+    # 使用mne模块创建一个RawArray对象，其中包含sample和info对象
+    raw = mne.io.RawArray(sample, info)
+    fig = raw.plot(duration=440, n_channels=30, color='black', scalings='auto')
+    fig.savefig(f"G:\脑电波\more_pic\\{cnt}.jpg")
+    # print(data['subject_ids'][cnt-1], data['text_ids'][cnt-1])
+    cnt = cnt + 1