Python多维列表计算矩阵对角线之和

可以使用 NumPy 库来计算多维列表的矩阵对角线之和。具体方法如下：

import numpy as np

# 创建一个3*3的矩阵
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 计算矩阵对角线之和
sum_of_diag = np.trace(matrix)

print(sum_of_diag)  # 输出：15

在上面的代码中，我们首先使用 NumPy 库中的 `array` 函数来创建一个3*3的矩阵，然后使用 `trace` 函数来计算矩阵的对角线之和，最后将结果打印出来。

相关问题

Python多维列表计算4*4矩阵对角线之和

可以使用以下代码来计算一个4x4矩阵的对角线之和:

matrix = [[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12], [13, 14, 15, 16]]

sum = 0
for i in range(len(matrix)):
    sum += matrix[i][i]

print("矩阵对角线之和为：", sum)

输出：

矩阵对角线之和为： 34

python将多维矩阵的主对角线各减去一个给定值代码

要将多维矩阵的主对角线各减去一个给定的值，可以使用NumPy库进行操作。下面是一个示例代码：

import numpy as np

# 创建一个4x4的矩阵
matrix = np.array([[1, 2, 3, 4],
                   [5, 6, 7, 8],
                   [9, 10, 11, 12],
                   [13, 14, 15, 16]])

# 要减去的值
value_to_subtract = 2

# 提取主对角线的索引
diagonal_indices = np.diag_indices(min(matrix.shape))

# 减去给定的值
matrix[diagonal_indices] -= value_to_subtract

print(matrix)

这段代码会输出各元素减去给定值后的矩阵：

[[-1  2  3  4]
 [ 5  4  7  8]
 [ 9 10  9 12]
 [13 14 15 14]]

你可以根据需要修改矩阵的大小、内容以及要减去的值。