本教程详细介绍了如何使用Python连接PostgreSQL数据库，并提供了执行基本CRUD（创建、读取、更新、删除）操作的示例代码。
Python 操作 PostgreSQL 数据库是一种常见的任务，在数据处理和Web应用开发中尤为常见。PostgreSQL 是一个功能强大的开源关系型数据库系统，而 Python 作为一种流行的编程语言，提供了多种方式来与之交互。在 Python 中，最常用且推荐的库是 `psycopg2` ，它允许程序员用Python代码执行SQL查询，并实现对数据库的连接、增删改查等操作。
要使用 `psycopg2` 模块，你需要确保已经正确安装了该模块。可以在命令行中运行以下命令来安装：`pip3 install psycopg2` 或者 `pip3 install psycopg2-binary` 。通常情况下，推荐使用 `psycopg2-binary` 版本，因为它包含了所有必要的依赖项。
连接到数据库是操作的起点。下面是一个简单的示例代码，展示了如何用 Python 连接到 PostgreSQL 数据库：
```python
import psycopg2
conn = psycopg2.connect(
database=testdb,
user=postgres,
password=pass123,
host=127.0.0.1,
port=5432
)
print(Opened database successfully)
```
这段代码创建了一个到名为 testdb 的数据库的连接，如果该数据库不存在，PostgreSQL 会自动创建。成功建立连接后，程序将打印 Opened database successfully。
在 PostgreSQL 数据库中创建表是基础操作之一。下面是一个 Python 示例代码片段，用于在 testdb 数据库中创建一个名为 COMPANY 的表格：
```python
import psycopg2
conn = psycopg2.connect(
database=testdb,
user=postgres,
password=pass123,
host=127.0.0.1,
port=5432
)
cur = conn.cursor()
cur.execute(
CREATE TABLE COMPANY (
ID INT PRIMARY KEY NOT NULL,
NAME TEXT NOT NULL,
AGE INT NOT NULL,
ADDRESS CHAR(50),
SALARY REAL
)
)
print(Table created successfully)
conn.commit()
conn.close()
```
此示例创建了一个包含员工信息的表格，包括ID（主键）、姓名、年龄、地址和薪水。
插入数据是通过调用游标的 `execute()` 方法完成。以下是如何向 COMPANY 表中添加记录的一个例子：
```python
import psycopg2
conn = psycopg2.connect(
database=testdb,
user=postgres,
password=pass123,
host=127.0.0.1,
port=5432
)
cur = conn.cursor()
# 插入多条记录
cur.execute(
INSERT INTO COMPANY (ID, NAME, AGE, ADDRESS, SALARY)
VALUES (1, Paul, 32, California, 20000.0);
)
cur.execute(
INSERT INTO COMPANY (ID, NAME, AGE, ADDRESS, SALARY)
VALUES (2, Allen, 25, Texas, 15000.0);
)
conn.commit()
print(Records inserted successfully)
conn.close()
```
更新和删除操作同样通过执行相应的SQL语句来完成。例如，要更新一条记录：
```python
cur.execute(
UPDATE COMPANY
SET SALARY = 25000.0
WHERE ID = 1;
)
```
或者删除一条记录：
```python
cur.execute(
DELETE FROM COMPANY
WHERE ID = 4;
)
```
在进行这些操作后，别忘了提交事务 (`conn.commit()`) 并关闭连接 (`conn.close()`).
通过 `psycopg2` 库，Python 提供了丰富的接口来管理和操作 PostgreSQL 数据库。无论是创建新表、插入数据、更新记录还是删除信息，都可以通过编写直观的 Python 代码轻松实现。掌握这些基本操作后，你可以构建复杂的数据处理逻辑，并为你的项目提供强大的数据支持。
5星
