编写一个自定义函数，使其能够返回传入的n×n矩阵的对角线元素之和。（如下图所示，若n为奇数，则中心元素<图中为元素5>只被计算一次）

以下是一个用Python编写的自定义函数，用于返回输入矩阵的对角线元素之和。函数名为`diag_sum`，输入为一个二维列表（n×n矩阵），返回值为一个整数，表示对角线元素的总和。

def diag_sum(matrix):
    n = len(matrix)
    diagonal_sum = 0

    # 遍历矩阵的每一行和每一列
    for i in range(n):
        for j in range(n):
            # 如果元素在矩阵的对角线上，则将其加到总和中
            if i == j:
                diagonal_sum += matrix[i][j]

    return diagonal_sum

使用示例：

# 创建一个3×3矩阵
matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

# 调用diag_sum函数，得到对角线元素之和
result = diag_sum(matrix)
print(result)  # 输出：25

注意：这个函数假设输入的矩阵是一个方阵（即行数和列数相等），并且只考虑对角线上的元素。如果输入的矩阵不是方阵，或者对角线上的元素不是整数，那么这个函数可能无法正常工作。在实际应用中，你可能需要添加一些额外的错误检查和处理代码来确保函数的正确性。

相关问题

使用matlab写一个自定义函数，使其能够返回传入的n×n矩阵的对角线元素之和。（如下图所示，若n为奇数，则中心元素<图中为元素5>只被计算一次）

可以使用下面的Matlab函数来计算n×n矩阵的对角线元素之和：

function sum = diagonal(A)
    n = size(A, 1);
    sum = 0;
    for i = 1:n
        sum = sum + A(i, i);
    end
end

这个函数接受一个矩阵作为输入，并返回该矩阵的对角线元素之和。首先，我们使用`size`函数获取矩阵的行数和列数，并将它们存储在变量`n`中。然后，我们使用一个循环遍历矩阵的每个元素，并使用`A(i, i)`来获取对角线上的元素。最后，我们将所有对角线元素相加并返回结果。

请注意，这个函数假设输入矩阵是一个方阵（即行数和列数相等）。如果输入矩阵不是方阵，函数将返回错误信息。此外，如果输入矩阵的行数或列数小于或等于零，函数也将返回错误信息。

使用这个函数的示例代码如下：

% 创建一个3×3矩阵
A = [1 2 3; 4 5 6; 7 8 9];

% 调用自定义函数计算对角线元素之和
sum = diagonal(A);

% 输出结果
disp(sum);  % 输出：25

在这个示例中，我们创建了一个3×3矩阵`A`，并使用`diagonal`函数计算其对角线元素之和。输出结果为25，因为该矩阵的对角线元素为1+5+9=15，加上中心元素7=8-1=7，总和为25。

使用matlab编写一个自定义函数，使得其能够对传入的非负整数n，生成杨辉三角的前n行。

当然可以，以下是一个使用MATLAB编写的自定义函数，该函数可以生成杨辉三角的前n行：

function triangle = generate_pascal_triangle(n)
    % 输入参数：n，非负整数
    % 输出参数：triangle，杨辉三角的前n行

    % 初始化杨辉三角的前n行
    triangle = zeros(n, n);

    % 生成第一行的杨辉三角
    triangle(1, 1) = 1;

    % 生成前n-1行的杨辉三角
    for i = 2:n
        triangle(i, 1) = triangle(i-1, 1);
        triangle(i, i) = 1;
        for j = 2:i-1
            triangle(i, j) = triangle(i-1, j-1) + triangle(i-1, j);
        end
    end
end

使用这个函数的方法如下：

% 调用函数生成杨辉三角的前5行
n = 5;
triangle = generate_pascal_triangle(n);

这将返回一个矩阵，矩阵的形状是 n x n，并且矩阵的前 n 行是生成的杨辉三角。请注意，此函数的每一行的数值均根据Pascal三角的规则进行计算。对于每一行的非主对角线元素，其值等于上一行相应元素和上一行上方元素之和。