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

一、前言

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

代码语言：javascript复制

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

二、实现过程

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

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

代码语言：javascript复制

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列表，方便后续处理。

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

三、总结

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

python 数据库 cursor 函数 优化

0 人点赞