From a2ec068b515ceb2fbca601645d80cb7227b90e4a Mon Sep 17 00:00:00 2001
From: hojob2003 <2916685612@qq.com>
Date: Thu, 29 Feb 2024 15:57:45 +0800
Subject: [PATCH] =?UTF-8?q?=E5=8F=AF=E8=A7=86=E5=8C=96=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 可视化.py | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)
 create mode 100644 可视化.py

diff --git a/可视化.py b/可视化.py
new file mode 100644
index 0000000..7a48950
--- /dev/null
+++ b/可视化.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
\ No newline at end of file