大家好,我是皮皮。
一、前言
前几天在Python最强王者交流群【刘苏秦】问了一个Python
数据库数据处理的问题,一起来看看吧。
cursor = connect.cursor()
sql = "SELECT * FROM infos where status='' "
cursor.execute(sql)
result=random.sample(cursor.fetchall(),10)
result=[dict(i).get('num')for i in result]
for i in result:
sql2 = f"DELETE FROM infos WHERE num = '{i}'"
cursor.execute(sql2)
connect.commit()
connect.close()
return result
有优化办法没,功能就是随机查询10个然后删除这10个?
二、实现过程
这里【莫生气】给了一个思路:
呀~小主,我帮你优化了一下代码,优化后的代码如下所示:
import random
def delete_random_records(connect):
cursor = connect.cursor()
sql = "SELECT * FROM infos WHERE status = ''"
cursor.execute(sql)
result = random.sample(cursor.fetchall(), 10)
result_ids = [dict(record).get('num') for record in result]
sql2 = f"DELETE FROM infos WHERE num IN ({','.join(['%s']*len(result_ids))})"
cursor.execute(sql2, result_ids)
connect.commit()
connect.close()
return result_ids
优化说明:
- 将导入的random库放到代码开头,遵循Python的惯例。
- 将代码封装成一个函数,提高代码的可重用性。
- 使用参数化查询,避免SQL注入的风险。
- 使用
IN
语句一次性删除多条记录,减少与数据库的交互次数。 - 返回删除的记录ID列表,方便后续处理。
顺利地解决了粉丝的问题。
三、总结
大家好,我是皮皮。这篇文章主要盘点了一个Python
数据库处理的问题,文中针对该问题,给出了具体的解析和代码实现,帮助粉丝顺利解决了问题。
最后感谢粉丝【刘苏秦】提问,感谢【莫生气】给出的思路和代码解析,感谢【懒人在思考】等人参与学习交流。
【提问补充】温馨提示,大家在群里提问的时候。可以注意下面几点:如果涉及到大文件数据,可以数据脱敏后,发点demo数据来(小文件的意思),然后贴点代码(可以复制的那种),记得发报错截图(截全)。代码不多的话,直接发代码文字即可,代码超过50行这样的话,发个.py文件就行。