如何编写SQL查询指定姓名的唯一ID并通过mysql-connect存入Python列表

1. 前置依赖安装 #

首先安装MySQL官方Python驱动：

pip install mysql-connector-python

2. 完整代码实现 #

import mysql.connector
from mysql.connector import Error

def get_jake_unique_ids():
    jake_ids = []
    connection = None
    cursor = None
    try:
        # 连接数据库，替换为你自己的数据库配置
        connection = mysql.connector.connect(
            host='你的数据库地址',
            user='你的数据库用户名',
            password='你的数据库密码',
            database='你的数据库名'
        )
        if connection.is_connected():
            cursor = connection.cursor()
            # SQL查询：获取姓名为Jake的所有唯一ID，用DISTINCT实现去重
            query = "SELECT DISTINCT ID FROM 你的表名 WHERE Name = %s"
            # 采用参数化查询，规避SQL注入风险
            cursor.execute(query, ('Jake',))
            # 获取所有查询结果，默认返回格式为[(id1,), (id2,), ...]
            results = cursor.fetchall()
            # 提取ID字段转为普通整数列表
            jake_ids = [row[0] for row in results]

    except Error as e:
        print(f"数据库操作出错: {e}")
    finally:
        # 关闭游标和连接，释放资源
        if cursor:
            cursor.close()
        if connection and connection.is_connected():
            connection.close()
    return jake_ids

# 调用函数获取结果
if __name__ == "__main__":
    jake_id_list = get_jake_unique_ids()
    print("Jake对应的唯一ID列表：", jake_id_list)

结果说明 #

你提供的样例表运行后，输出的列表为：[145928, 987468, 897426, 448961]

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