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;
关注以下公众号可以更方便查看文章哦(* ̄︶ ̄)