用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中进行数据库操作了。
