526互联

使用Python随机查询数据库中10个信息然后删除这10个信息

发布时间 2023-10-24 22:06:34作者: dcpeng

大家好，我是皮皮。

一、前言

前几天在Python最强王者交流群【刘苏秦】问了一个Python数据库数据处理的问题，一起来看看吧。

image.png

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个？

image.png

二、实现过程

这里【莫生气】给了一个思路：

呀~小主，我帮你优化了一下代码，优化后的代码如下所示：

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

优化说明：

  1. 将导入的random库放到代码开头，遵循Python的惯例。
  2. 将代码封装成一个函数，提高代码的可重用性。
  3. 使用参数化查询，避免SQL注入的风险。
  4. 使用IN语句一次性删除多条记录，减少与数据库的交互次数。
  5. 返回删除的记录ID列表，方便后续处理。

顺利地解决了粉丝的问题。

image.png

三、总结

大家好，我是皮皮。这篇文章主要盘点了一个Python数据库处理的问题，文中针对该问题，给出了具体的解析和代码实现，帮助粉丝顺利解决了问题。

最后感谢粉丝【刘苏秦】提问，感谢【莫生气】给出的思路和代码解析，感谢【懒人在思考】等人参与学习交流。

【提问补充】温馨提示，大家在群里提问的时候。可以注意下面几点：如果涉及到大文件数据，可以数据脱敏后，发点demo数据来（小文件的意思），然后贴点代码（可以复制的那种），记得发报错截图（截全）。代码不多的话，直接发代码文字即可，代码超过50行这样的话，发个.py文件就行。

image.png