Python中用psycopg2模块操作PostgreSQL方法

当我们需要与PostgreSQL数据库进行交互时，Python中psycopg2模块是一个不错的选择。以下是用psycopg2模块连接、创建和查询PostgreSQL数据库的完整攻略：

安装psycopg2模块

使用psycopg2模块需要先安装。你可以在终端使用如下命令安装：

pip install psycopg2

连接PostgreSQL数据库

连接PostgreSQL数据库需要指定数据库名、用户名、密码、主机名和端口号。你可以使用下面的代码来连接数据库：

import psycopg2

conn = psycopg2.connect(
    database="database_name",
    user="user_name",
    password="password",
    host="host_name",
    port="port_number"
)

如果连接成功，将返回一个连接对象conn。

创建表和插入数据

在创建表之前应该先创建一个游标对象cur：

cur = conn.cursor()

使用游标对象在数据库中执行SQL语句：

cur.execute('''CREATE TABLE test (id INTEGER PRIMARY KEY, name TEXT);''')
cur.execute('''INSERT INTO test (id, name) VALUES (1, 'Alice');''')
cur.execute('''INSERT INTO test (id, name) VALUES (2, 'Bob');''')

查询数据

使用游标对象执行查询语句：

cur.execute('''SELECT * FROM test;''')

rows = cur.fetchall()

for row in rows:
    print(row)

运行上述代码将输出以下结果：

(1, 'Alice')
(2, 'Bob')

关闭连接和游标对象

最后需要关闭连接和游标对象：

cur.close()
conn.close()

以上就是Python中用psycopg2模块操作PostgreSQL的完整攻略。下面是另一个示例，演示如何使用psycopg2模块连接数据库并创建一个简单的表。

import psycopg2

# 连接到PostgreSQL数据库
conn = psycopg2.connect(
    database="database_name",
    user="user_name",
    password="password",
    host="host_name",
    port="port_number"
)

# 创建一个游标对象
cur = conn.cursor()

# 创建一个名为students的表
cur.execute('''CREATE TABLE students (id SERIAL PRIMARY KEY, name TEXT, age INTEGER);''')

# 在表中插入数据
cur.execute('''INSERT INTO students (name, age) VALUES ('Alice', 20);''')
cur.execute('''INSERT INTO students (name, age) VALUES ('Bob', 21);''')

# 查询表中所有数据
cur.execute('''SELECT * FROM students;''')
rows = cur.fetchall()

for row in rows:
    print(row)

# 关闭游标和连接对象
cur.close()
conn.close()

输出结果如下：

(1, 'Alice', 20)
(2, 'Bob', 21)