需求:小计,用python链接数据库+从本地导入excel文件入库
首先是建立数据库
下面会建立一个名为mydatabase的数据库,可以在你的sql控制台看到。
import mysql.connector
mydb = mysql.connector.connect(
host='localhost',
user='root',
password='123456'
)
mycursor = mydb.cursor()
# 创建数据库
mycursor.execute('CREATE DATABASE mydatabase')
# 切换到新创建的数据库
mycursor.execute('USE mydatabase')
# 创建表
mycursor.execute('CREATE TABLE mytable (id INT AUTO_INCREMENT PRIMARY KEY, name VARCHAR(255), age INT);')
# 提交事务
mydb.commit()
# 关闭数据库连接
mydb.close()
下面是在这个数据库中新建一张表,并写入内容。
# 导包
import pymysql
# 连接mysql
conn = pymysql.connect(host='localhost', port=3306, user='root', password='123456', db='mydatabase')
# 执行语句
cursor = conn.cursor()
sql = 'insert into mytable(id,name,age) values(%s,%s,%s)'
rows = cursor.executemany(sql, [('13', '小李', 21), ('10', '小王', 22), ('11', '小猪', 23), ('12', '小狗', 24)])
conn.commit()
cursor.close()
好的以上就是简单的创建数据库和数据库中的表,但是我们绝大部分人都是为了直接导入excel数据,像dbheaver中虽然有直接导入csv文件的方式,但是调试起来很麻烦。那么我们在学会了上面的方式后,来尝试下从本地文件传入excel文件进入我们的mysql数据库中。
以下是片段代码,最终完整代码我会放在最后
import pandas as pd
import os
from sqlalchemy import create_engine
#输入你的数据库登录信息方便后期调用
root = 'root' # MySQL用户名
password = '123456' # MySQL密码
localhost = 'localhost' # MySQL服务器地址
port = '3306' # MySQL端口号
mydatabase = 'mydatabase' # 数据库名称
这块,把你的数据信息放入,这样方便你后期的调用。
# 读取Excel文件
outfolder_path = r"F:\今天"
output_name = "00"
output_path = os.path.join(outfolder_path, output_name + ".xlsx")
df = pd.read_excel(output_path, engine='openpyxl')
# 删除列名前后的空格
df.columns = df.columns.str.strip()
# 与你的数据库建立连接
engine = create_engine(f'mysql+pymysql://{root}:{password}@{localhost}:{port}/{mydatabase}')
此段代码放入你的excel文件路径,以及你的文件名称。数据库不允许表头有空格字符串,所以要删除列名前后的空格。然后将数据赋值给你的数据库。
# 将数据从pandas数据帧导入到MySQL数据库
table_name = 'example_table'
df.to_sql(table_name, engine, if_exists='replace', index=False)
此处为写入数据库。
以下为全部代码:
import pandas as pd
import os
from sqlalchemy import create_engine
#输入你的数据库登录信息方便后期调用
root = 'root' # MySQL用户名
password = '123456' # MySQL密码
localhost = 'localhost' # MySQL服务器地址
port = '3306' # MySQL端口号
mydatabase = 'mydatabase' # 数据库名称
# 读取Excel文件
outfolder_path = r"F:\今天"
output_name = "00"
output_path = os.path.join(outfolder_path, output_name + ".xlsx")
df = pd.read_excel(output_path, engine='openpyxl')
# 删除列名前后的空格
df.columns = df.columns.str.strip()
# 与你的数据库建立连接
engine = create_engine(f'mysql+pymysql://{root}:{password}@{localhost}:{port}/{mydatabase}')
# 将数据从pandas数据帧导入到MySQL数据库
table_name = 'example_table'
df.to_sql(table_name, engine, if_exists='replace', index=False)
以上是今天分享,祝大家生活工作愉快