Python 如何将数据导入 PostgreSQL 数据库

在数据处理与管理领域，将数据从 Python 导入到 PostgreSQL 数据库是一项常见且重要的任务。PostgreSQL 作为一款强大的开源关系型数据库，具备高可靠性和丰富的数据类型支持，而 Python 凭借其简洁语法和丰富的库，为数据导入操作提供了便利的环境。

我们需要安装必要的库。psycopg2 是 Python 中用于连接和操作 PostgreSQL 数据库的常用库。可以使用 pip install psycopg2 命令进行安装。安装完成后，就可以开始编写代码实现数据导入。

在导入数据前，要先建立与 PostgreSQL 数据库的连接。示例代码如下：

import psycopg2

try:
    connection = psycopg2.connect(
        user="your_username",
        password="your_password",
        host="your_host",
        port="your_port",
        database="your_database"
    )
    cursor = connection.cursor()
    print("成功连接到数据库")
except (Exception, psycopg2.Error) as error:
    print("连接数据库时出现错误:", error)

这里，我们通过 psycopg2.connect 方法，传入用户名、密码、主机、端口和数据库名来建立连接，并创建游标对象用于执行 SQL 语句。

假设我们有一个包含数据的列表，例如 data = [('John', 25), ('Jane', 30)]，要将这些数据导入到名为 users 的表中，表结构为 (name VARCHAR(50), age INT)，可以使用如下代码：

insert_query = "INSERT INTO users (name, age) VALUES (%s, %s)"
for row in data:
    cursor.execute(insert_query, row)
connection.commit()

这段代码使用 INSERT INTO 语句将数据逐行插入到 users 表中，最后通过 connection.commit() 方法提交事务，确保数据被真正写入数据库。

如果数据量较大，逐行插入效率较低。此时可以使用 executemany 方法，示例如下：

cursor.executemany(insert_query, data)
connection.commit()

这样一次操作就能插入多条数据，大大提高了导入效率。

完成数据导入后，不要忘记关闭游标和数据库连接，释放资源：

cursor.close()
connection.close()
print("数据库连接已关闭")

通过以上步骤，我们就能利用 Python 高效地将数据导入到 PostgreSQL 数据库中，为后续的数据处理和分析奠定坚实基础。