Pandas连接MySQL数据库

pandas是一个强大的Python工具包,能够快速帮助我们做很多数据处理。但是在利用pandas连接数据库时,也会遇到很多问题,在此我总结了一个相对较为简单的连接范式,供大家参考学习。

先上代码:

python 复制代码
import pandas as pd

# 数据库参数
MYSQL_USER = 'root'
MYSQL_PASS = '****' # 你的密码
MYSQL_HOST = "localhost" # 你的host
MYSQL_PORT = 3306
MYSQL_CHARSET = 'utf8mb4' # 数据库编码格式

# 数据库名称
MYSQL_DATABASE = "system"

import pymysql
from sqlalchemy import create_engine

# url
MYSQL_DATABASE_URL = fr"mysql+pymysql://{MYSQL_USER}:{MYSQL_PASS}@{MYSQL_HOST}:{MYSQL_PORT}/{MYSQL_DATABASE}?charset={MYSQL_CHARSET}"
# 创建引擎
engine = create_engine(MYSQL_DATABASE_URL)

# 数据库调用函数
def access_db(sql, data=None, is_many=False):
    '''
    :param sql: sql执行语句
    :param data: 数据,如果没有数据则执行sql代码
    :param is_many: 是否为多条数据
    :return: 查询数据
    '''
    conn = pymysql.connect(host=MYSQL_HOST, port=MYSQL_PORT, user=MYSQL_USER,
                         password=MYSQL_PASS, db=MYSQL_DATABASE, charset=MYSQL_CHARSET)
    cur = conn.cursor()

    if is_many:
        try:
            cur.executemany(sql, data)
            conn.commit()
        except Exception as e:
            print(e)
            conn.rollback()
    elif data:
        try:
            cur.execute(sql, data)
            conn.commit()
        except Exception as e:
            print(e)
            conn.rollback()
    else:
        cur.execute(sql)
        exe_data = cur.fetchall()
        conn.commit()
        return exe_data
    cur.close()
    conn.close()

# 查询数据
sql = "select * from user"
df_user = pd.read_sql(sql=sql, con=engine)
print(df_user)

上述代码中用到的函数,在查询语句,插入语句(一次可插入一条或多条)等方面都可以直接调用,但需要确保sql语句的正确性,注意更改数据库参数为你自己需要连接的数据库,sql建议在代码调用前最好直接在数据库进行过查询并能够获取到正确的数据再放入代码层,避免因为一些基本语法问题导致问题定位浪费时间和精力。

相关推荐
siriuuus5 分钟前
MySQL 的 MyISAM 与 InnoDB 存储引擎的核心区别
mysql·1024程序员节
Q16849645151 小时前
提高命令行运行效率-正则 表达式
数据库·mysql
Pluchon2 小时前
硅基计划5.0 MySQL 陆 视图&JDBC编程&用户权限控制
数据库·mysql·1024程序员节
Thepatterraining5 小时前
MySQL数据存储黑科技:Page布局、行存储与压缩算法全解密
数据库·mysql
IT教程资源D6 小时前
[N_149]基于微信小程序网上商城系统
mysql·vue·前后端分离·springboot网上商城·网上商城小程序
洲覆6 小时前
SQL 性能优化:出现 sql 比较慢怎么办?
开发语言·数据库·sql·mysql
想不明白的过度思考者6 小时前
MySQL 8.0.x 全平台安装指南:Windows、CentOS、Ubuntu 详细步骤与问题解决
windows·mysql·centos
weixin_307779136 小时前
Linux 下 Docker 与 ClickHouse 的安装配置及 MySQL 数据同步指南
linux·数据库·mysql·clickhouse·运维开发
程序新视界8 小时前
一张图解析MySQL中InnoDB的逻辑存储结构
数据库·后端·mysql
一只叫煤球的猫16 小时前
MySQL 索引的 “最左前缀原则”,用查字典的例子讲透
数据库·mysql·性能优化