Python生成三对角矩阵

最新推荐文章于 2024-07-04 04:24:22 发布

SVLAB

最新推荐文章于 2024-07-04 04:24:22 发布

阅读量1.7k

点赞数 4

文章标签： python 矩阵 numpy

本文链接：https://blog.youkuaiyun.com/SVLAB/article/details/128944087

版权

该代码定义了一个名为ThreediaMatrix的函数，它使用numpy库生成一个具有指定下、中、上对角线元素的三对角矩阵。输入参数包括对角线的值和矩阵的大小，返回的结果是一个5x5的三对角矩阵。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import numpy as np
def ThreediaMatrix(down,mid,up,n):         
    array_a=np.diag([down]*(n-1),-1)    # down是对角线下一行的元素，末尾是-1，n-1是元素的个数
    array_b = np.diag([mid] * n)        # mid是对角线上的元素，默认为0
    array_c=np.diag([up]*(n-1),1)       # up是对角线上面的元素，末尾是1
    matrix_A=array_a+array_b+array_c
    return  matrix_A
A=ThreediaMatrix(-1,2,1,5)
print(A)

[[ 2  1  0  0  0]
 [-1  2  1  0  0]
 [ 0 -1  2  1  0]
 [ 0  0 -1  2  1]
 [ 0  0  0 -1  2]]