import sqlite3 import json import random # 根据distance查询数据 def get_data_by_distance(distance, one=True): conn = sqlite3.connect('database.db') cursor = conn.cursor() cursor.execute(f"SELECT * FROM Tracker WHERE distance = {distance}") rows = cursor.fetchall() if one: random_index = random.randint(0, len(rows) - 1) return json.loads(rows[random_index][2]) return [json.loads(row[2]) for row in rows] mouse_data = get_data_by_distance(100) for data in mouse_data: print(data) # cursor.execute("SELECT * FROM Tracker") # rows = cursor.fetchall() # index_list = [] # for index, distance, mouse_tracker in rows: # if not json.loads(mouse_tracker): # index_list.append(index) # print(f"Index: {index}, Distance: {distance}") # # print(f"Index: {index}, Distance: {distance}") # # print(f"Mouse tracker: {json.loads(mouse_tracker)}") # # 根据 index 删除数据 # # for index in index_list: # # cursor.execute(f"DELETE FROM Tracker WHERE id = {index}") # # conn.commit() # # # 关闭连接 # conn.close() # print(index_list)