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()