Python操作PostgreSQL数据库
个人简介：一个从会计转行数据分析师的三旬老汉
擅长领域：数据分析、数据仓库、大数据
博客内容：平时会将自己工作中遇到的问题进行归纳总结，分享给各位小伙伴，意在帮助大家少加班、不掉发，让我们相互学习，一起进步。
文章目录
- 前言
- 一、连接PostgreSQL数据库
- 二、增
-
- （一）建表
- （二）插入数据
- 三、删
-
- （一）删表
- （二）删除表数据
- 三、改
-
- （一）更新数据
- （二）结果展示
- 四、查
-
- （一）查询数据库表
- 总结
前言
本文分享使用Python操作PostgreSQL数据库的基本方法，包括数据库连接、增、删、改、查，供各位小伙伴参考。
一、连接PostgreSQL数据库
操作MySQL数据库主要使用psycopg2包，连接PostgreSQL数据库的语法为connect(IP, 端口, 用户名, 密码, 数据库名,编码格式)。具体代码如下：
import psycopg2
conn = psycopg2.connect(
host='xxx',
port='xxx',
dbname='xxx',
user='xxx',
password='xxx'
)
二、增
（一）建表
以下语法中1、2、4、5为操作数据库的通用语法，下文其他步骤也会使用。
- 创建mysql连接：pymysql.connect()，详见上文第一点
- 创建游标：conn.cursor()
- 编写建表语句，建表语句写在三引号中：
（1）create_sql = “”“xxx”“”
（2）cursor.execute(create_sql) - 提交语句：conn.commit()
- 关闭mysql连接：conn.close()
import psycopg2
conn = psycopg2.connect(
host='xxx',
port='xxx',
dbname='xxx',
user='xxx',
password='xxx'
)
cursor = conn.cursor()
# 建表
create_sql = """
CREATE TABLE xxx.yyy(
id int
,name varchar(10)
);
"""
cursor.execute(create_sql)
print("create successfully")
conn.commit()
conn.close()
（二）插入数据
- 插入单条数据的语法和sql的插入类似，直接键值对插入，插入语句写在双引号中。具体代码如下：
import psycopg2
conn = psycopg2.connect(
host='xxx',
port='xxx',
dbname='xxx',
user='xxx',
password='xxx'
)
cursor = conn.cursor()
# 插入
insert_sql = "INSERT INTO xx.yyy (id, name) VALUES (%s, '%s')" % (1, '张三')
cursor.execute(insert_sql)
print("insert successfully")
conn.commit()
conn.close()
- 查询插入数据，适用于从数据库其他表查询关联后插入数据到新表。插入语句写在三引号中。具体代码如下：
import psycopg2
conn = psycopg2.connect(
host='xxx',
port='xxx',
dbname='xxx',
user='xxx',
password='xxx'
)
cursor = conn.cursor()
# 插入
insert_sql = """
DROP TABLE IF EXISTS xx.yyy;
CREATE TABLE IF NOT EXISTS xx.yyy AS
SELECT
aa
,bb
FROM xx.yyy
"""
cursor.execute(insert_sql)
print("insert successfully")
conn.commit()
conn.close()
三、删
（一）删表
直接删除表数据和表结构，其语法和SQL删表语法一致。具体代码如下：
import psycopg2
conn = psycopg2.connect(
host='xxx',
port='xxx',
dbname='xxx',
user='xxx',
password='xxx'
)
cursor = conn.cursor()
# 删表
cursor.execute("DROP TABLE IF EXISTS xx.yyy")
conn.commit()
conn.close()
（二）删除表数据
只删除数据，不删除表结构。其语法和SQL删表数据语法一致。具体代码如下：
import psycopg2
conn = psycopg2.connect(
host='xxx',
port='xxx',
dbname='xxx',
user='xxx',
password='xxx'
)
cursor = conn.cursor()
# 删表数据
cursor.execute("TRUCATE TABLE xx.yyy")
conn.commit()
conn.close()
三、改
（一）更新数据
将上文所建的yyy表中，“张三”改为“何老六”。具体代码如下：
import psycopg2
conn = psycopg2.connect(
host='xxx',
port='xxx',
dbname='xxx',
user='xxx',
password='xxx'
)
cursor = conn.cursor()
cursor.execute("UPDATE xx.yyy SET name = '何老六' WHERE id = 1")
conn.commit()
conn.close()
（二）结果展示
四、查
（一）查询数据库表
直接使用pandas包查询数据库表，语法为read_sql(“select xxx from yyy”, con=数据库连接)。
import psycopg2
import pandas as pd
conn = psycopg2.connect(
host='xxx',
port='xxx',
dbname='xxx',
user='xxx',
password='xxx'
)
df = pd.read_sql("select * from xx.yyy limit 100;", con=conn)
print(df)
总结
除了PostgreSQL数据库外，业界常用的数据库MySQL也可以使用Python进行操作，具体情况可参考作者的另一篇博客Python操作MySQL数据库。