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建议在代码调用前最好直接在数据库进行过查询并能够获取到正确的数据再放入代码层,避免因为一些基本语法问题导致问题定位浪费时间和精力。

相关推荐
PiscesCanon13 小时前
达梦开启awr功能报错:[-7160]:Object [DBMS_WORKLOAD_REPOSITORY_DATA_LOW] is invalid
数据库·mysql
冻咸鱼14 小时前
数据库操作
数据库·mysql·数据库操作
duqiao_wang15 小时前
Mysql数据库系统库数据恢复
数据库·mysql
计算机学姐15 小时前
基于微信小程序的垃圾分类管理系统【2026最新】
java·vue.js·spring boot·mysql·微信小程序·小程序·mybatis
Albert Edison18 小时前
【MySQL】数据类型
数据库·mysql·adb·oracle
遇见火星18 小时前
MYSQL-物理备份(xtrabackup)使用指南
数据库·mysql·adb
huihuihuanhuan.xin19 小时前
后端八股之mysql
数据库·mysql
洋不写bug19 小时前
数据库数据类型,数据值类型,字符串类型,日期类型详解
数据库·mysql
Java水解21 小时前
MySQL 中 ROW_NUMBER() 函数详解
后端·mysql
知其然亦知其所以然1 天前
面试官问:MySQL表损坏怎么修?不会这三招你就凉了!
后端·mysql·面试