python连接数据库

python连接数据库的方法：

• 导入sqlalchemy包
• 导入psycopg2包

一、利用sqlalchemy包连接数据库

import os
import sys
import pandas as pd
import datetime
from sqlalchemy import create_engine #python操作数据库法1
from string import Template
#初始化引擎
engine=create_engine('postgresql+psycopg2://pg_username :pg_password@pg_host:pg_port/pg_database')
sql1="""select * from  $arg1 where is_fi ='1' limit 100 """
sql1=Template(sql1)## template方法,初始化一个字符串，调用substitute或safe_subsititute，将key值与方法中传递过来的参数对应上
data=pd.read_sql_query(sql1.substitute(arg1='xx.dm_user_xxxx_202006'),engine)#利用pandas读取数据,substitute指定字符串的参数

二、利用psycopg2包连接数据库

import psycopg2 #python操作数据库法2
#初始化
conn=psycopg2.connect(database="xx",user="xx",password="xx",host="xx",port="xxxx")
cursor=conn.cursor()
cursor.execute("select * from xx.dm_user_xxxx_202006 where is_fi ='1'limit 100") #执行语句
rows = cursor.fetchall()#将返回所有结果，返回二维元组;cursor.fetchone()：将只取最上面的第一条结果
data = pd.DataFrame(rows)#转化成数据框的格式

三、python数据导入数据库
使用pd.io.sql.to_sql函数将数据导入数据库，其中if_exists的参数
replace的意思，如果表存在，删了表，再建立一个新表，把数据插入
append的意思，如果表存在，把数据插入，如果表不存在创建一个表！！
fail的意思如果表存在，什么也不做。

pd.io.sql.to_sql(data,'cx_rk',engine,schema='tmp',if_exists='replace',index=False) #数据导入到数据库

四、两个包的区别
psycopg2是PostgreSQL官方指定推荐包，对于PostgreSQL来说，他是最合适的包；而sqlalchemy可以在很多类型的数据库使用，包含PostgreSQL；

​
关注以下公众号可以更方便查看文章哦(*￣︶￣)

​