如何通过对3D数组中的每个值求和来返回2D数组

def sum_3d_array(arr):
    # 获取3D数组的维度大小
    depth = len(arr)
    rows = len(arr[0])
    cols = len(arr[0][0])

    # 创建一个空的2D数组
    result = [[0] * cols for _ in range(rows)]

    # 遍历3D数组的每个元素，并将其值加到对应位置的2D数组中
    for d in range(depth):
        for r in range(rows):
            for c in range(cols):
                result[r][c] += arr[d][r][c]

    # 返回求和后的2D数组
    return result