计算二维数组对角线和

好的，计算二维数组对角线和的代码如下：

def diagonal_sum(arr):
    n = len(arr)
    sum = 0
    for i in range(n):
        for j in range(n):
            if i == j:
                sum += arr[i][j]
    return sum

其中，`arr` 表示输入的二维数组，`n` 表示数组的维度，`sum` 表示对角线元素的和。通过两重循环遍历数组，当行列相等时，说明该元素是对角线上的元素，将其加入到 `sum` 中。最后返回 `sum` 即可。

请注意，这段代码只适用于正方形的二维数组。如果输入的二维数组不是正方形，那么该函数会抛出 `IndexError` 异常。