如何创建一个Python函数来操作两个或多个相同形状的嵌套列表？

def operate_lists(list1, list2):
    # 检查形状是否相同
    if len(list1) != len(list2) or any(len(sublist1) != len(sublist2) for sublist1, sublist2 in zip(list1, list2)):
        return "Error: The shape of the lists is not the same."

    # 创建新的列表
    result = []

    # 遍历两个列表的元素
    for i in range(len(list1)):
        sublist_result = []
        for j in range(len(list1[i])):
            # 对应位置的元素相加
            element_result = list1[i][j] + list2[i][j]
            sublist_result.append(element_result)
        result.append(sublist_result)

    return result