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

用Python函数操作数据库:连接、查询、插入和删除数据

发布时间:2023-07-01 01:01:40

在Python中,我们可以通过使用数据库模块来连接、查询、插入和删除数据。Python的标准库中提供了多个数据库模块,如sqlite3、MySQLdb、psycopg2等。在本文中,我们以sqlite3为例来介绍如何使用Python函数操作数据库。

首先,我们需要导入sqlite3模块并连接数据库。sqlite3是一个轻量级的数据库模块,适用于小规模的数据库操作。我们可以使用connect()函数来连接数据库,并传入数据库文件的路径作为参数。

import sqlite3

# 连接数据库
conn = sqlite3.connect('database.db')

接下来,我们可以使用execute()函数来执行SQL查询和插入操作。我们可以通过传入SQL语句作为参数来执行查询操作,通过fetchall()函数来获取查询结果。同样地,我们可以传入SQL语句和参数来执行插入操作。

# 执行查询操作
cursor = conn.execute("SELECT * FROM users")
rows = cursor.fetchall()    # 获取查询结果

# 执行插入操作
conn.execute("INSERT INTO users (name, age) VALUES (?, ?)", ('John', 25))
conn.commit()    # 提交事务

在数据库操作完成后,我们需要通过调用close()函数来关闭数据库连接。

# 关闭数据库连接
conn.close()

除了查询和插入操作,我们也可以使用execute()函数来执行其他SQL操作,如删除、更新和创建表等。

# 执行删除操作
conn.execute("DELETE FROM users WHERE name = 'John'")
conn.commit()

# 执行更新操作
conn.execute("UPDATE users SET age = 30 WHERE name = 'Tom'")
conn.commit()

# 执行创建表操作
conn.execute("CREATE TABLE users (id int, name text, age int)")
conn.commit()

以上就是使用Python函数操作数据库的基本过程。我们可以根据实际情况,通过编写SQL语句和调用数据库模块提供的函数来完成不同的数据库操作。这样,我们就可以方便地在Python中进行数据库操作了。