PostgreSQL - 使用Python将数据插入表中

本文将介绍如何使用Python编写代码将数据插入PostgreSQL数据库中的表。我们将使用psycopg2库来连接到数据库并执行SQL语句。首先，确保已经安装了postgresql和psycopg2。

连接到PostgreSQL数据库

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

import psycopg2

try:
    conn = psycopg2.connect(
        database='mydatabase',
        user='myusername',
        password='mypassword',
        host='localhost',
        port='5432'
    )

    print("Connected to database")
except:
    print("Unable to connect to database")

替换上述代码中的数据库名称、用户名、密码、主机和端口为正确的值。

创建表

在插入数据之前，我们需要创建一个表来表示数据的结构。在此示例中，我们将创建一个名为employee的表，其中包含id、name和age字段。使用以下Python代码创建表：

cur = conn.cursor()

cur.execute('''
    CREATE TABLE employee
    (id INT PRIMARY KEY NOT NULL,
    name TEXT NOT NULL,
    age INT NOT NULL);
''')

conn.commit()

print("Table created successfully")

在上面的代码执行完成后，我们将在PostgreSQL数据库中创建一个employee表。

插入数据

现在我们已经连接到数据库并创建了表，我们将向employee表中插入一些数据。使用以下Python代码将数据插入表中：

cur = conn.cursor()

cur.execute("INSERT INTO employee (id, name, age) VALUES (%s, %s, %s)", (1, 'John Doe', 35))
cur.execute("INSERT INTO employee (id, name, age) VALUES (%s, %s, %s)", (2, 'Mike Smith', 25))
cur.execute("INSERT INTO employee (id, name, age) VALUES (%s, %s, %s)", (3, 'Jane Jones', 42))

conn.commit()

print("Records created successfully")

在上述代码中，我们将三条数据插入到employee表中。每条数据都由id、name和age字段组成。使用占位符%s将值插入到SQL语句中，并在元组中提供实际值。

每个插入操作都由“cur.execute()”函数执行，然后使用“conn.commit()”提交更改。

关闭连接

完成所有操作后，使用以下Python代码关闭与PostgreSQL数据库的连接：

conn.close()

print("Connection closed")

以上是将数据插入PostgreSQL数据库表的Python代码。您可以根据需要更改代码来适应您的特定需求。