以下是一个保存用户角色分配的代码示例：

# 假设有一个角色分配的字典，key 是用户ID，value 是角色列表
user_roles = {
    1: ['admin', 'user'],
    2: ['user'],
    3: ['admin'],
    4: ['user']
}

# 保存用户角色分配的函数
def save_user_roles(user_id, roles):
    # 更新角色分配字典
    user_roles[user_id] = roles
    
    # 在这里可以添加保存到数据库或文件的逻辑
    # 例如：将字典转换为 JSON 格式并保存到文件
    import json
    with open('user_roles.json', 'w') as file:
        json.dump(user_roles, file)

# 调用保存用户角色分配函数
save_user_roles(1, ['admin', 'user'])

在这个示例中，我们使用一个字典 user_roles 来保存用户角色分配信息。user_roles 的 key 是用户ID，value 是一个角色列表。我们定义了一个 save_user_roles 函数，用于更新角色分配字典，并且在函数内部可以添加保存到数据库或文件的逻辑。

在示例中，我们使用了 json 模块将角色分配字典转换为 JSON 格式，并保存到名为 user_roles.json 的文件中。你可以根据实际需求，选择适合的保存方式。