聊聊 Python 数据处理全家桶(PgSQL篇)

开发 后端
PgSQL,全称为 PostgreSQL,是一款免费开源的关系型数据库。相比最流行的 Mysql 数据库,PgSQL 在可靠性、数据完整性、扩展性方面具有绝对的优势

 [[416887]]

本文转载自微信公众号「AirPython」,作者星安果。转载本文请联系AirPython公众号。

1. 前言

PgSQL,全称为 PostgreSQL,是一款免费开源的关系型数据库

相比最流行的 Mysql 数据库,PgSQL 在可靠性、数据完整性、扩展性方面具有绝对的优势

本篇文章将聊聊如何使用 Python 操作 PgSQL 数据库

2. PgSQL 使用

Python 操作 PgSQL,需要先安装依赖包「 psycopg2 」

# 安装依赖包 
pip3 install psycopg2 
  • 1.
  • 2.

接下来,就可以使用 Python 来操作数据库了

2-1 数据库连接及游标对象

使用 psycopg2 中的「 connect() 」方法连接数据库,创建数据库连接对象及游标对象

import psycopg2 
 
# 获得连接对象 
database:数据库名称 
user:用户名 
password:密码 
# host:数据库ip地址 
# port:端口号,默认为5432 
conn = psycopg2.connect(database="db_name"user="postgres"password="pwd", host="127.0.0.1", port="5432"
 
# 获取游标对象 
cursor = conn.cursor() 
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.
  • 11.
  • 12.

获取游标对象后,就可以执行 SQL,进而操作数据库了

2-2 插入数据

首先,编写插入数据的 SQL 语句及参数( 可选 )

# 构建SQL语句 
# 方式一:直带参数 
 sql = "INSERT INTO student (name,age) \ 
                        VALUES (%s, '%s')" % \ 
          ('xag',23) 
 
# 方式二:参数分离 
sql = """INSERT INTO student (name,age) VALUES (%s, %s)""" 
# 参数 
params = ('xag',23) 
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.

然后,使用游标对象执行 SQL

# 执行sql 
# 注意:params可选,根据上面的参数方式来选择设置 
cursor.execute(sql,[params]) 
  • 1.
  • 2.
  • 3.

接着,使用连接对象提交事务

# 事务提交 
onn.commit() 
  • 1.
  • 2.

最后,释放游标对象及数据库连接对象

# 释放游标对象及数据库连接对象 
cursor.close() 
conn.close() 
  • 1.
  • 2.
  • 3.

2-3 查询数据

游标对象的 fetchone()、fetchmany(size)、fetchall() 这 3个函数即可以实现单条数据查询、多条数据查询、全部数据查询

# 获取一条记录 
one_data = cursor.fetchone() 
print(one_data) 
 
# 获取2条记录 
many_data = cursor.fetchmany(2) 
print(many_data) 
 
# 获取全部数据 
all_data = cursor.fetchall() 
print(all_data) 
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.
  • 11.

需要注意的是,条件查询与上面的插入操作类似,条件语句可以将参数分离出来

# 条件查询 SQL语句 
sql = """SELECT * FROM student where id = %s;""" 
 
# 对应参数,参数结尾以逗号结尾 
params = (1,) 
 
# 执行SQL 
cursor.execute(sql, params) 
 
# 获取所有数据 
datas = cursor.fetchall() 
print(datas) 
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.
  • 11.
  • 12.

2-4 更新数据

更新操作和上面操作一样,唯一不同的是,执行完 SQL 后,需要使用连接对象提交事务,才能将数据真实更新到数据库中

def update_one(conn, cursor): 
    """更新操作""" 
    # 更新语句 
    sql = """update student set name = %s where id = %s  """ 
    params = ('AirPython', 1,) 
 
    # 执行语句 
    cursor.execute(sql, params) 
 
    # 事务提交 
    conn.commit() 
 
    # 关闭数据库连接 
    cursor.close() 
    conn.close() 
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.
  • 11.
  • 12.
  • 13.
  • 14.
  • 15.

2-5 删除数据

删除数据同更新数据操作类似

def delete_one(conn, cursor): 
    """删除操作""" 
    # 语句及参数 
    sql = """delete from  student where id = %s  """ 
    params = (1,) 
 
    # 执行语句 
    cursor.execute(sql, params) 
 
    # 事物提交 
    conn.commit() 
 
    # 关闭数据库连接 
    cursor.close() 
    conn.close() 
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.
  • 11.
  • 12.
  • 13.
  • 14.
  • 15.

3. 最后

通过上面操作,可以发现 Python 操作 PgSQl 与 Mysql 类似,但是在原生 SQL 编写上两者还是有很多差异性

更多关于 PgSQL 的操作可以参考下面链接:

https://www.postgresql.org/docs/14/index.html

 

责任编辑:武晓燕 来源: AirPython
相关推荐

2020-09-27 08:36:21

Python Memcached缓存

2020-09-29 08:35:08

MongoDBPython数据

2020-09-24 06:49:34

PythonRedis

2023-07-31 08:21:22

语法校对器Pick

2024-05-15 09:53:22

2022-07-15 08:45:07

slotVue3

2022-02-09 15:23:41

大数据流计算Spark

2017-09-10 17:41:39

React全家桶单元测试前端测试

2017-09-13 15:05:10

React前端单元测试

2024-01-31 23:22:35

vaexPython

2021-03-16 10:12:24

python内置函数

2023-09-25 13:19:41

pandasPython

2023-12-12 11:06:37

PythonPandas数据

2020-11-02 15:56:04

大数据数据库技术

2018-12-07 14:50:35

大数据数据采集数据库

2017-07-21 14:22:17

大数据大数据平台数据处理

2025-01-27 12:19:51

2023-09-25 10:16:44

Python编程

2021-07-20 15:37:37

数据开发大数据Spark
点赞
收藏

51CTO技术栈公众号