Python中如何将变量与JSON文件内的字符串ID匹配并执行对应操作

你原有代码存在逻辑冗余和判断错误的问题：不需要遍历整个JSON的键，直接用Python字典的in运算符就能判断指定键是否存在，原有判断if str(member.id) in data[str(member.id)]的逻辑是判断用户ID是否存在于对应键的值里，和你要的匹配键的需求不符。

完整实现步骤 #

• 第一步：读取JSON文件加载为字典
• 第二步：直接判断转换为字符串的member.id是否在字典的键集合中
• 第三步：匹配成功后执行自定义操作（比如删除对应条目、加角色等）
• 第四步：如果修改了字典内容，需要把更新后的内容写回JSON文件

可直接运行的代码示例 #

import json

# 读取JSON文件
with open('jsons/mutes.json', 'r', encoding='utf-8') as data_file:
    data = json.load(data_file)

user_id = str(member.id)
# 检查用户ID是否是JSON的键
if user_id in data:
    # 匹配成功，执行自定义操作
    await member.add_roles(role)
    # 如果要删除对应条目，执行下面的代码
    del data[user_id]
    # 其他自定义操作可以在此处补充

# 如果修改了data字典，需要把更新后的内容写回JSON文件
with open('jsons/mutes.json', 'w', encoding='utf-8') as data_file:
    json.dump(data, data_file, indent=4, ensure_ascii=False)

代码说明 #

• 用user_id in data判断键是否存在的时间复杂度是O(1)，比遍历所有键的*O(n)*效率更高
• 写入JSON时加indent=4可以保持文件的格式化可读性，ensure_ascii=False避免特殊字符乱码
• 如果你不需要保留修改后的JSON，可以跳过写回文件的步骤

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