求4*4整型数组的主对角线元素的和，请编fun函数

您好，以下是Python的fun函数实现：

def fun(arr):
    length = len(arr)
    diagonal_sum = 0
    for i in range(length):
        diagonal_sum += arr[i][i]
    return diagonal_sum

其中，arr为4*4的整型数组，length为数组长度。在for循环中，我们使用了变量i迭代数组的每一行，累加对角线上的元素，并返回最终的对角线元素和。