替换任意嵌套列表中的元素

def replace_element(nested_list, old_element, new_element):
    if isinstance(nested_list, list):  # 检查是否为列表
        for i, element in enumerate(nested_list):
            if isinstance(element, list):  # 检查是否为嵌套列表
                nested_list[i] = replace_element(element, old_element, new_element)
            elif element == old_element:
                nested_list[i] = new_element
    return nested_list

nested_list = [1, [2, [3, 4], 5], 6, [7, 8]]
old_element = 3
new_element = 'A'

result = replace_element(nested_list, old_element, new_element)
print(result)

[1, [2, ['A', 4], 5], 6, [7, 8]]