Python PostgreSQL - 插入数据

Python PostgreSQL是一个用于连接和操作PostgreSQL数据库的Python库。在本文中，我们将介绍如何使用Python PostgreSQL库来插入数据到数据库中。

安装依赖库

在开始之前，我们需要安装psycopg2库来连接和操作PostgreSQL数据库。可以使用以下命令来安装：

pip install psycopg2

连接到数据库

在插入数据之前，我们首先需要连接到PostgreSQL数据库。可以使用以下代码来连接到数据库：

import psycopg2

# 建立数据库连接
conn = psycopg2.connect(database="your_database", user="your_username", password="your_password", host="your_host", port="your_port")

# 创建游标
cur = conn.cursor()

确保将your_database替换为实际的数据库名，your_username替换为实际的用户名，your_password替换为实际的密码，your_host替换为实际的主机名，your_port替换为实际的端口号。

插入数据

连接到数据库后，我们可以使用下面的代码来插入数据到数据库中：

# 插入单条数据
query = "INSERT INTO your_table (column1, column2, column3) VALUES (%s, %s, %s)"
data = ("value1", "value2", "value3")
cur.execute(query, data)

# 插入多条数据
query = "INSERT INTO your_table (column1, column2, column3) VALUES (%s, %s, %s), (%s, %s, %s)"
data = [("value1", "value2", "value3"), ("value4", "value5", "value6")]
cur.executemany(query, data)

# 提交事务
conn.commit()

确保将your_table替换为实际的表名，并根据表结构和数据类型提供正确的数据。

关闭连接

在插入完数据后，我们应该关闭数据库连接，释放资源。可以使用以下代码来关闭连接：

# 关闭游标
cur.close()

# 关闭连接
conn.close()

示例

下面是一个完整的使用Python PostgreSQL插入数据的示例代码：

import psycopg2

# 建立数据库连接
conn = psycopg2.connect(database="your_database", user="your_username", password="your_password", host="your_host", port="your_port")

# 创建游标
cur = conn.cursor()

# 插入单条数据
query = "INSERT INTO your_table (column1, column2, column3) VALUES (%s, %s, %s)"
data = ("value1", "value2", "value3")
cur.execute(query, data)

# 插入多条数据
query = "INSERT INTO your_table (column1, column2, column3) VALUES (%s, %s, %s), (%s, %s, %s)"
data = [("value1", "value2", "value3"), ("value4", "value5", "value6")]
cur.executemany(query, data)

# 提交事务
conn.commit()

# 关闭游标
cur.close()

# 关闭连接
conn.close()

确保将示例代码中的占位符替换为实际的值，并根据你的需求进行相应更改。

希望本文能帮助你开始使用Python PostgreSQL插入数据，更多关于Python操作PostgreSQL的信息可以参考官方文档psycopg2 documentation。