python连接数据库

1、python连接sqlite3，anaconda默认已经安装了sqlite3

# 拓展：链接sql数据库
import sqlite3
import pandas as pd

# import pymysql  默认已经安装了sqlite3
con = sqlite3.connect("test.db")
con.execute("drop table if exists person")
con.commit()

# 创建数据表
con.execute("create table person(id int primary key, name varchar(30), age int)")

# 向表中插入数据
con.execute("insert into person(id, name ,age) values(22, 'ls', 17)")
con.execute("insert into person(id, name ,age) values(33, 'zs', 18)")

# 从数据库中读取数据，sql指定查询的数据（用来构造DataFrame）。con数据库的链接。
t = pd.read_sql("select id, name, age from person", con)

display(t)

 

2、python连接mysql，连接之前需要先安装mysqlclient

import MySQLdb
import pandas as pd

# 打开数据库连接，输入账号密码（假设账号密码都是root）
con = MySQLdb.connect(host = 'localhost', port = 3306, 
                     user = 'root', passwd ='root', 
                     db = 'db01', charset='utf8' )

# 写好sql语句
sql = "select id, name, age from zt_test"
df = pd.read_sql(sql, con)
display(df.head())