Python编写的数据库操作案例:连接数据库并执行SQL查询
发布时间:2023-12-04 19:37:36
Python编写的数据库操作案例:
Python提供了多种数据库操作的模块,其中最常用的是sqlite3和pymysql模块。下面以sqlite3为例子,介绍如何连接数据库并执行SQL查询。
首先,需要安装sqlite3模块。可以使用以下命令进行安装:
pip install sqlite3
接下来,可以使用以下代码连接数据库,并执行SQL查询:
import sqlite3
# 连接数据库
conn = sqlite3.connect('example.db')
# 创建一个游标对象
cursor = conn.cursor()
# 创建一个表
cursor.execute('''CREATE TABLE IF NOT EXISTS stocks
(date text, trans text, symbol text, qty real, price real)''')
# 插入一条数据
cursor.execute("INSERT INTO stocks VALUES (?, ?, ?, ?, ?)", ('2022-01-05', 'BUY', 'RHAT', 100, 35.14))
# 提交事务
conn.commit()
# 执行查询
cursor.execute("SELECT * FROM stocks")
# 获取查询结果
result = cursor.fetchall()
# 输出查询结果
for row in result:
print(row)
# 关闭连接
conn.close()
上述代码中,首先通过connect函数连接数据库。如果数据库文件不存在,将会自动创建一个新的数据库文件。然后,创建一个游标对象用于执行SQL语句。接着,使用execute方法执行SQL语句,可以是创建表、插入数据、更新数据、删除数据等操作。最后,使用fetchall方法获取查询结果,并使用循环遍历输出查询结果。最后,使用close方法关闭数据库连接。
这是一个简单的数据库操作案例,可以根据实际需求,使用更复杂的SQL语句进行数据库操作。
使用sqlite3模块连接数据库和执行SQL查询的例子就是这样。另外,pymysql模块的用法也和sqlite3类似,只需要将引入的模块名改为pymysql即可。具体使用方法可以参考官方文档或相关教程。
