使用pymysql对mysql进行数据插入

前言

在写上位机代码时候，需要将采集的数据存入数据库中。这里进行了简单的研究（因为表涉及2张，此外创建表也研究了会）特记录下。

这里有组数据需要存入表中，表结构为一个主表一个从表

2张表的DDL

-- AS01.sensor_test definition
​
CREATE TABLE `sensor_test` (
  `id` int(11) NOT NULL AUTO_INCREMENT,
  `test_mode` varchar(100) NOT NULL,
  `test_time` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
  PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=3 DEFAULT CHARSET=utf8;
​
-- AS01.test_database definition
​
CREATE TABLE `test_database` (
  `id` int(11) NOT NULL AUTO_INCREMENT,
  `pid` int(11) NOT NULL,
  `angle` float DEFAULT NULL,
  `voltage` float DEFAULT NULL,
  PRIMARY KEY (`id`),
  KEY `test_database_FK` (`pid`),
  CONSTRAINT `test_database_FK` FOREIGN KEY (`pid`) REFERENCES `sensor_test` (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=5 DEFAULT CHARSET=utf8;
​

python语句

import pymysql
conn = pymysql.connect(# 这里需要填入你自己数据的信息
    host='your_host_name',
    user='your_user_name',
    password='your_password', 
    database='AS01' # 操作的表名
)
​
cursor = conn.cursor()
​
test_mode = '5'
sql = "INSERT INTO sensor_test (test_mode) VALUES (%s)"
val = [test_mode]
cursor.execute(sql, val)
​
# 获取新插入的数据的id
new_id = cursor.lastrowid
​
# 插入数据到从表
data = [[1, 2.3], [2, 3.3], [3, 4.4]]
for row in data:
    sql = "INSERT INTO test_database (pid, angle, voltage) VALUES (%s, %s, %s)"
    val = [new_id] + row
    cursor.execute(sql, val)
​
# 提交更改并关闭连接
conn.commit()
cursor.close()
conn.close()

运行后，数据库数据如下：

简单测试下：

SELECT * FROM AS01.test_database WHERE pid = 3 

PS：
存入后，为了方便其他客户查看数据，需要把存入的数据保存成表格使用

import pandas as pd
import pymysql

# 连接数据库
conn = pymysql.connect(
    host='localhost',
    user='root',
    password='123456',
    database='AS01'
)

# 查询数据
sql = "SELECT * FROM AS01.test_database WHERE pid = 3"
df = pd.read_sql(sql, conn)

# 写入CSV文件
df.to_csv('data3.csv', index=False)

# 关闭连接
conn.close()