Python | PostgreSQL 中的数据库管理

在 Python 中，我们可以使用 psycopg2 库来管理 PostgreSQL 数据库。这个库提供了许多方法来执行 SQL 语句并与 PostgreSQL 数据库进行交互。

安装 psycopg2

在使用 psycopg2 之前，需要先安装它。可以使用 pip 命令进行安装：

pip install psycopg2-binary

连接到 PostgreSQL

使用以下代码来连接到 PostgreSQL 数据库：

import psycopg2

conn = psycopg2.connect(
    host="localhost",
    database="mydatabase",
    user="myusername",
    password="mypassword"
)

在 connect() 方法中传递了数据库连接参数，并将连接对象保存在变量 conn 中。

执行 SQL 语句

使用 cursor() 方法获得一个游标对象，并使用该对象执行 SQL 语句：

cur = conn.cursor()
cur.execute("SELECT * FROM mytable")
rows = cur.fetchall()

在这个示例中，我们将查询 mytable 表中的所有行，并将结果存储在 rows 变量中。

插入数据

使用以下代码将数据插入 PostgreSQL 数据库：

cur = conn.cursor()
cur.execute("INSERT INTO mytable (name, age) VALUES (%s, %s)", ("Alice", 25))
conn.commit()

在这个示例中，我们将 name 和 age 插入到 mytable 表中。在执行完插入操作后，使用 commit() 方法提交事务。

关闭连接

在完成与 PostgreSQL 数据库的交互后，使用以下代码关闭连接：

cur.close()
conn.close()

这将关闭游标对象和数据库连接对象。

以上是关于 Python 和 PostgreSQL 中的数据库管理的介绍。