我有一个数据库,有五个column:项目名称、1月、2月、3月、4月。我会生成一个prj变量以及一个mons变量。prj是一个字符,mons是一个列表,列表里随机出现1月、2月、3月、4月的排列组合。我想查询项目名称为prj,column为mons列表中的所有数据。我用的库是pysql,给我代码,
时间: 2024-03-16 09:45:09 浏览: 52
2010年3月vf上机题库
好的,那么这里提供使用 pysql 库的代码,来查询项目名称为 prj,column 为 mons 列表中的所有数据:
```python
import pysql
# 连接到数据库
conn = pysql.connect('your_database.db')
# 生成测试数据
prj = 'project1'
mons = ['1月', '2月', '4月']
# 构造 SQL 语句
sql = 'SELECT * FROM budget_table WHERE 项目名称=? AND ({})'.format(' OR '.join(['{}=?'.format(m) for m in mons]))
# 执行 SQL 语句
cursor = conn.cursor()
cursor.execute(sql, [prj] + mons)
results = cursor.fetchall()
# 打印查询结果
for row in results:
print(row)
# 关闭数据库连接
conn.close()
```
这段代码与之前的代码非常相似,只是将 `sqlite3` 库替换为了 `pysql` 库。其余部分的操作流程与之前的代码是一致的。
阅读全文