使用Python将数据存储到PostgreSQL数据库中

2023-04-04 00:00:00 python 数据存储数据库中

首先，需要在Python中安装psycopg2包，可用以下命令进行安装：

pip install psycopg2

然后可以使用以下方式连接到PostgreSQL数据库：

import psycopg2

conn = psycopg2.connect(
    host="localhost",
    database="mydatabase",
    user="myusername",
    password="mypasswd"
)

其中，host、database、user、password分别为要连接的PostgreSQL数据库的主机名、数据库名、用户名和密码。

接下来，可以创建一个游标对象，并执行SQL语句，将数据存储到PostgreSQL数据库中：

cur = conn.cursor()

sql = """
    insert into mytable (name, age) values ('pidancode.com', '20');
"""
cur.execute(sql)

sql = """
    insert into mytable (name, age) values ('皮蛋编程', '22');
"""
cur.execute(sql)

conn.commit()  # 提交事务

cur.close()
conn.close()

上述代码示例中，我们在mytable表中插入了两条记录，每条记录包括一个字符串类型的name字段和一个整型的age字段。

最后，需要调用conn.commit()方法提交事务，同时关闭游标和数据库连接对象。

相关文章