如何用Python在MySQL中基于二维数组删除多行？查询ID遇空列表

直接导致空列表的原因 #

• Cursor结果被覆盖：循环里每次调用mycursor.execute()都会重置cursor的结果集，最终cursor仅保留最后一次查询的结果。如果最后一次查询无匹配数据，IDtoDelete就会是空列表。
• 未及时读取查询结果：循环结束后才遍历cursor，只能拿到最后一次查询的结果，前面所有查询的结果都已丢失。

优化后的解决方案 #

方案1：修复原有循环逻辑（不推荐，效率低下） #

若要保留循环查询的方式，需在每次执行查询后立即读取结果并添加到列表：

IDtoDelete = []
for item in toDelete:
    sql = "SELECT id FROM schedulev1 WHERE CIF_train_uid = %s AND start_date = %s AND CIF_stp_indicator = %s"
    mycursor.execute(sql, item)
    # 每次查询后立即获取结果
    results = mycursor.fetchall()
    IDtoDelete.extend([x[0] for x in results])

方案2：批量查询（推荐，大幅提升效率） #

针对2万条数据的处理需求，循环单条查询会严重拖慢速度，用IN子句构造批量查询语句更高效：

# 构造IN子句的占位符，每个三元组对应一组(%s,%s,%s)
placeholders = ", ".join(["(%s, %s, %s)"] * len(toDelete))
# 扁平化二维数组，将所有参数整合到一个列表中
params = [item for sublist in toDelete for item in sublist]

sql = f"SELECT id FROM schedulev1 WHERE (CIF_train_uid, start_date, CIF_stp_indicator) IN ({placeholders})"
mycursor.execute(sql, params)
# 获取所有结果并提取id
IDtoDelete = [x[0] for x in mycursor.fetchall()]

额外注意事项 #

• 先验证数据库中是否存在匹配toDelete数组的记录：取数组内一条数据单独执行查询，确认能查到对应id。
• 批量查询时需注意数据库的参数数量限制，若toDelete长度过大（如超过1万条），可分成多个批次执行，避免触发参数上限。

内容的提问来源于stack exchange，提问作者big_bongos