pymysql——从数据库获取数据

连接数据库

import pymysql
connect = pymysql.connect(
  host="localhost",
  user="root",
  passwd="123456",
  database='shuju'
)
cursor = connect.cursor()
cursor.execute("select * from test")

cursor.fetchone()  # 单条查询结果

 

cursor.fetchmany(3)  # 查询指定条数结果

 

cursor.fetchall()  # 全量查询结果

 

cursor.close() #关闭游标
connect.close() #关闭与数据库的链接

 

import pymysql
class MysqlReader(object):
    def __init__(self):
        self.connect = pymysql.connect(host="localhost",user="root",passwd="123456",database='shuju')
    def getData(self):
        cursor = self.connect.cursor()
        cursor.execute("select * from test")
        res = cursor.fetchall()
        cursor.close()
        self.connect.close()
        return res
# 类调用
MR = MysqlReader()
res = MR.getData()
res = [list(i) for i in res]
df = pd.DataFrame(res, columns=range(len(res[0])))  # columns也可自定义