### Python批量替换模型文件中的键名 在Python中处理JSON或字典类型的模型文件时，可以通过编写一个通用函数来实现批量替换键名的功能。此功能不仅适用于单层字典，还可以扩展到嵌套字典的情况。 以下是基于需求编写的代码示例： ```python import json def replace_keys_in_dict(data, key_mapping): """ 替换单个字典中的键名。 参数: data (dict): 输入的字典数据。 key_mapping (dict): 键映射关系，旧键 -> 新键。 返回: dict: 修改后的字典。 """ if isinstance(data, dict): new_data = {} for key, value in data.items(): # 如果当前键需要被替换，则使用新键 new_key = key_mapping.get(key, key) new_value = replace_keys_in_dict(value, key_mapping) # 处理嵌套情况 new_data[new_key] = new_value return new_data elif isinstance(data, list): # 如果是列表，则递归处理其中的每一项 return [replace_keys_in_dict(item, key_mapping) for item in data] else: # 非字典/列表类型则直接返回原值 return data def batch_replace_keys_in_model_files(file_paths, key_mapping): """ 批量替換多个模型文件中的键名。 参数: file_paths (list): JSON文件路径列表。 key_mapping (dict): 键映射关系，旧键 -> 新键。 """ for file_path in file_paths: try: with open(file_path, 'r', encoding='utf-8') as f: model_data = json.load(f) updated_data = replace_keys_in_dict(model_data, key_mapping) with open(file_path, 'w', encoding='utf-8') as f: json.dump(updated_data, f, ensure_ascii=False, indent=4) print(f"{file_path} 的键已成功更新.") except Exception as e: print(f"处理 {file_path} 时发生错误: {e}") # 示例用法 if __name__ == "__main__": files_to_update = ['model1.json', 'model2.json'] # 待更新的模型文件列表 keys_to_rename = {'old_key': 'new_key', 'another_old_key': 'another_new_key'} # 键映射 batch_replace_keys_in_model_files(files_to_update, keys_to_rename) ``` #### 解释 1. **`replace_keys_in_dict` 函数**: 此函数用于递归地遍历字典及其子节点（如果存在），并根据 `key_mapping` 将匹配的键替换为新的键[^1]。它支持嵌套字典和列表结构，因此能够适应复杂的JSON模型文件[^2]。 2. **`batch_replace_keys_in_model_files` 函数**: 这是一个批处理函数，接受一组JSON文件路径以及键映射表作为参数。对于每个文件，先加载其内容，再调用 `replace_keys_in_dict` 更新键名，最后将修改后的内容重新写回文件[^1]。 3. **异常处理**: 在实际应用中可能会遇到各种意外情况（如文件不存在、格式不正确等）。通过捕获异常并打印错误消息，可以提高程序的健壮性[^1]。 --- ### 注意事项 - 确保目标文件确实是合法的JSON格式，否则可能引发解析错误。 - 对于大型文件，建议分步测试以验证逻辑无误后再运行整个流程。 - 若需保留原始文件备份，可以在覆盖前复制一份副本。 ---