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

前端皮皮

发布于 2023-10-25 09:36:00

1250

发布于 2023-10-25 09:36:00

文章被收录于专栏：前端进阶学习交流前端进阶学习交流

一、前言

前几天在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

优化说明：