Python 如何连接并运行SQL查询到PostgreSQL数据库

PostgreSQL是一个流行的开源关系型数据库管理系统，以其强大、可扩展和先进的特性而闻名。Python为与PostgreSQL数据库交互提供了出色的支持，允许开发人员无缝连接并执行SQL查询。在本文中，我们将探讨从Python连接到PostgreSQL数据库、运行SQL查询和检索结果的逐步过程。我们将提供代码示例、解释和样本输出，以帮助您掌握这一基本技能。

在本文中，我们将学习如何使用Python连接并运行SQL查询到PostgreSQL数据库。

步骤1：安装psycopg2模块

第一步是安装psycopg2模块。这个模块为Python提供了与PostgreSQL的绑定。

考虑下面显示的命令。

命令

pip install psycopg2

步骤2：创建与数据库的连接

接下来的步骤是创建与数据库的连接。可以使用psycopg2.connect()函数来完成这一步。

考虑下面的代码片段：

代码片段

import psycopg2

conn = psycopg2.connect(
    host="localhost",
    dbname="my_database",
    user="postgres",
    password="my_password"
)

步骤3：创建游标

游标用于执行对数据库的SQL语句。要创建游标，可以使用连接对象的cursor()方法。

考虑下面所示的代码片段。

代码片段

cur = conn.cursor()

步骤4：执行SQL语句

一旦你有一个游标，你可以对数据库执行SQL语句。为了做到这一点，你可以使用游标对象的execute()方法。

考虑下面显示的代码片段。

代码片段

cur.execute("CREATE TABLE users (id SERIAL PRIMARY KEY, name TEXT)")

此语句创建了一个名为users的表，包含两个列：id和name。其中，id列是一个整型主键，name列是一个文本列。

步骤5：向数据库插入数据

创建表后，可以向其中插入数据。为此，可以使用游标对象的insert()方法。

考虑下方的代码片段。

代码片段

cur.execute("INSERT INTO users (name) VALUES ('John Doe')")

此语句将在用户表中插入一行数据，姓名为”John Doe”。

步骤6：查询数据库

一旦您将数据插入到数据库中，您可以对其进行查询。为了实现这一点，您可以使用光标对象的execute()方法。

考虑下面显示的代码片段。

代码片段

cur.execute("SELECT * FROM users")

这个语句从用户表中选择所有行。

步骤7：获取结果

execute() 方法返回一个结果集。要获取结果，可以使用 cursor 对象的 fetchall() 方法。

考虑下面的代码片段。

代码片段

results = cur.fetchall()

结果变量将包含结果集中所有行的列表。

步骤8：关闭连接

完成与数据库的操作后，应关闭连接。要做到这一点，您可以使用连接对象的close()方法。

请考虑下面显示的代码片段。

代码片段

conn.close()

输出

代码的输出将是用户表中所有行的列表。

[(1, 'John Doe')]

解释

• PostgreSQL是一种流行的关系数据库管理系统（RDBMS）。

• 要连接到PostgreSQL数据库，可以使用Python中的psycopg2模块。

• psycopg2模块为PostgreSQL提供了Python绑定。

• 要创建到PostgreSQL数据库的连接，可以使用psycopg2.connect()函数。

• 要创建游标，可以使用连接对象的cursor()方法。

• 要执行SQL语句，可以使用游标对象的execute()方法。

• 要向数据库中插入数据，可以使用游标对象的insert()方法。

• 要查询数据库，可以使用游标对象的execute()方法。

• 要获取结果，可以使用游标对象的fetchall()方法。

• 要关闭连接，可以使用连接对象的close()方法。

总结

在本文中，您学习了如何从Python连接并运行SQL查询到PostgreSQL数据库。您可以使用这些知识在Python中存储和检索PostgreSQL数据库中的数据。