欢迎访问宙启技术站
智能推送

如何在Python中使用函数实现数据库的连接与查询?

发布时间:2023-06-26 07:52:11

在Python中,我们可以使用Python自带的sqlite3模块来操作数据库。该模块可以连接各种常见的关系型数据库,如MySQL、PostgreSQL、Oracle等。

在使用该模块前,需要先安装Python的sqlite3模块。在绝大多数情况下,该模块已经默认安装,因此无需再次安装。

使用sqlite3模块连接数据库的一般步骤为:

1. 导入模块

在Python中使用sqlite3模块,首先需要将其导入。

import sqlite3

2. 连接数据库

使用sqlite3模块连接数据库的方法为:

con = sqlite3.connect('数据库名')

其中,'数据库名'为存放数据库文件的文件路径。如果数据库不存在,则会创建一个新的数据库文件。

3. 创建游标对象

使用连接对象创建游标对象,用于执行具体的数据库操作。方法如下:

cur = con.cursor()

4. 执行查询操作

使用游标对象执行SQL语句,从数据库中查询数据。例如:

cur.execute('SELECT * FROM student')

5. 获取查询结果

使用游标对象的fetch方法获取查询结果。例如:

result = cur.fetchall()

6. 关闭连接

使用连接对象的close方法关闭与数据库的连接。例如:

con.close()

整合以上步骤,我们可以编写一个Python程序来查询数据库,并输出查询结果。例如:

import sqlite3

con = sqlite3.connect('test.db')
cur = con.cursor()

cur.execute('SELECT * FROM student')
result = cur.fetchall()

for row in result:
    print(row)

con.close()

以上程序会连接名为'test.db'的数据库,并查询student表中的数据。查询结果将被存储在变量result中,程序会将结果输出到控制台,并最终关闭数据库连接。

需要注意的是,以上程序仅适用于SQLite数据库。如果要连接其他数据库,需要使用不同的Python模块,并修改连接方法和执行SQL语句的方式。但是,无论使用哪个模块,以上步骤和基本思路都是类似的。