📜  将Python列表插入 PostgreSQL 数据库(1)

📅  最后修改于: 2023-12-03 14:53:46.236000             🧑  作者: Mango

将Python列表插入 PostgreSQL 数据库

在Python中,有时需要将一个列表插入到PostgreSQL数据库中。以下是一个简单的介绍,如何使用Python的Psycopg2库将一个列表插入到PostgreSQL数据库中。

准备工作

在开始之前,确保已经安装了Psycopg2库并且PostgreSQL数据库在本地安装并运行。

pip install psycopg2

在Python中使用psycopg2连接到PostgreSQL数据库:

import psycopg2

conn = psycopg2.connect(
    host="localhost",
    database="testdb",
    user="postgres",
    password="password"
)
插入列表

现在,我们已经成功连接到PostgreSQL数据库,可以将接下来的代码粘贴到Python脚本中,用于将列表插入到数据库的表中。

import psycopg2

try:
    conn = psycopg2.connect(
        host="localhost",
        database="testdb",
        user="postgres",
        password="password"
    )

    cur = conn.cursor()

    # 创建表
    cur.execute('''CREATE TABLE IF NOT EXISTS users 
               (ID SERIAL PRIMARY KEY,
                NAME TEXT NOT NULL,
                AGE INT NOT NULL);''')

    # 要插入的数据
    users = [("John", 25), ("Alice", 30), ("Bob", 35)]

    # 将数据插入到表中
    for user in users:
        cur.execute("INSERT INTO users (NAME, AGE) VALUES (%s, %s)", user)

    conn.commit()

except (Exception, psycopg2.DatabaseError) as error:
    print(error)

finally:
    if conn is not None:
        conn.close()
        print('Database connection closed.')

此代码将在数据库中创建名为“users”的表,并将列表“users”中的数据插入到该表中。

结论

Psycopg2是一个功能强大的Python库,可用于将列表或其他类型的数据插入到PostgreSQL数据库中。通过使用相应的数据库连接详细信息,即可轻松连接到PostgreSQL数据库,使用标准SQL语法,将数据存储到表中。