PostgreSQL psycopg2: 一次查询插入多行数据

在本文中，我们将介绍如何使用PostgreSQL数据库和psycopg2库在一次查询中插入多行数据。psycopg2是一个Python库，用于与PostgreSQL数据库进行交互和操作。

插入单行数据

在开始讲解如何插入多行数据之前，我们先回顾一下如何使用psycopg2在一次查询中插入单行数据。假设我们有一个名为”customers”的表，包含”customer_id”、”customer_name”和”customer_address”三列。我们要向该表中插入一行数据。

首先，我们需要连接到PostgreSQL数据库。假设我们已经安装了psycopg2库，并且有一个可以访问的PostgreSQL数据库。以下是连接到数据库并插入数据的示例代码：

import psycopg2

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

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

# 插入数据
cur.execute("INSERT INTO customers (customer_id, customer_name, customer_address) VALUES (1, 'Alice', '123 Main St')")

# 提交事务
conn.commit()

# 关闭游标和数据库连接
cur.close()
conn.close()

在上述代码中，我们首先使用psycopg2库的connect方法连接到PostgreSQL数据库。在cur.execute方法中，我们执行了一个SQL INSERT语句，将数据插入到”customers”表中。最后，通过调用conn.commit()方法提交事务。

插入多行数据

要在一次查询中插入多行数据，我们可以使用execute_values方法。execute_values方法是psycopg2库提供的一个方便的方法，用于插入多个值。以下是如何使用execute_values方法插入多行数据的示例代码：

import psycopg2
from psycopg2.extras import execute_values

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

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

# 多行数据
rows = [
    (2, 'Bob', '456 Elm St'),
    (3, 'Charlie', '789 Oak St'),
    (4, 'David', '321 Pine St')
]

# 插入多行数据
execute_values(cur, "INSERT INTO customers (customer_id, customer_name, customer_address) VALUES %s", rows)

# 提交事务
conn.commit()

# 关闭游标和数据库连接
cur.close()
conn.close()

在上述代码中，我们首先将要插入的多行数据定义为一个列表，列表中的每个元素都是一个包含要插入的值的元组。然后，我们使用execute_values方法将这些多行数据插入到”customers”表中。

值得注意的是，我们在执行execute_values之前需要导入execute_values方法。导入方法的方式是通过from psycopg2.extras import execute_values语句导入。这样，我们就可以直接使用execute_values方法。