Python操作PostgreSQL数据库

个人简介：一个从会计转行数据分析师的三旬老汉
擅长领域：数据分析、数据仓库、大数据
博客内容：平时会将自己工作中遇到的问题进行归纳总结，分享给各位小伙伴，意在帮助大家少加班、不掉发，让我们相互学习，一起进步。

文章目录

• 前言
• 一、连接PostgreSQL数据库
• 二、增
• • （一）建表
  • （二）插入数据
• 三、删
• • （一）删表
  • （二）删除表数据
• 三、改
• • （一）更新数据
  • （二）结果展示
• 四、查
• • （一）查询数据库表
• 总结

---

前言

本文分享使用Python操作PostgreSQL数据库的基本方法，包括数据库连接、增、删、改、查，供各位小伙伴参考。

---

一、连接PostgreSQL数据库

操作MySQL数据库主要使用psycopg2包，连接PostgreSQL数据库的语法为connect(IP, 端口, 用户名, 密码, 数据库名,编码格式)。具体代码如下：

import psycopg2conn = psycopg2.connect(    host='xxx',    port='xxx',    dbname='xxx',    user='xxx',    password='xxx')

二、增

（一）建表

以下语法中1、2、4、5为操作数据库的通用语法，下文其他步骤也会使用。

1. 创建mysql连接：pymysql.connect()，详见上文第一点
2. 创建游标：conn.cursor()
3. 编写建表语句，建表语句写在三引号中：
   （1）create_sql = “”“xxx”“”
   （2）cursor.execute(create_sql)
4. 提交语句：conn.commit()
5. 关闭mysql连接：conn.close()

import psycopg2conn = 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()

（二）插入数据

1. 插入单条数据的语法和sql的插入类似，直接键值对插入，插入语句写在双引号中。具体代码如下：

import psycopg2conn = 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()

2. 查询插入数据，适用于从数据库其他表查询关联后插入数据到新表。插入语句写在三引号中。具体代码如下：

import psycopg2conn = 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 psycopg2conn = 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 psycopg2conn = 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 psycopg2conn = 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 psycopg2import pandas as pdconn = 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数据库。

来源地址：https://blog.csdn.net/zhangliushi/article/details/130388281