You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
39 lines
1.0 KiB
39 lines
1.0 KiB
1 year ago
|
import json
|
||
|
|
|
||
|
|
import numpy as np
|
||
|
|
import mysql.connector
|
||
|
|
import json
|
||
|
|
|
||
|
|
# 连接到MySQL数据库
|
||
|
|
conn = mysql.connector.connect(
|
||
|
|
host="localhost",
|
||
|
|
user="root",
|
||
|
|
password="123456",
|
||
|
|
port=3306,
|
||
|
|
database="脑电波数据库"
|
||
|
|
)
|
||
|
|
|
||
|
|
# 获取游标
|
||
|
|
cursor = conn.cursor()
|
||
|
|
|
||
|
|
data = np.load("ABMD_results3.npy", allow_pickle=True).item()
|
||
|
|
point = data['data']
|
||
|
|
labels = data['labels']
|
||
|
|
predict = data['predict']
|
||
|
|
text = data['text']
|
||
|
|
subject_ids = data['subject_ids']
|
||
|
|
text_ids = data['text_ids']
|
||
|
|
for i in range(3000):
|
||
|
|
idd = i + 12000
|
||
|
|
point2list = point[i].tolist()
|
||
|
|
pointJson = json.dumps({'data': point2list})
|
||
|
|
predict2list = predict[i].tolist()
|
||
|
|
predictJson = json.dumps({'predict': predict2list})
|
||
|
|
item = (idd, pointJson, int(labels[i]), predictJson, str(text[i]), int(subject_ids[i]), int(text_ids[i]))
|
||
|
|
cursor.execute("INSERT INTO naodianbo (id, data, labels, predict, text, subject_ids, text_ids) VALUES (%s, %s, %s, %s, %s, %s, %s)", item)
|
||
|
|
|
||
|
|
# 提交更改
|
||
|
|
conn.commit()
|
||
|
|
# 关闭游标和连接
|
||
|
|
cursor.close()
|
||
|
|
conn.close()
|