📅  最后修改于: 2023-12-03 14:53:46.236000             🧑  作者: Mango
在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语法,将数据存储到表中。