想生成如下的矩阵,主对角线是2,对角线两侧平行的是1,运用python的numpy比较简单。
[[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]]
numpy提供两种生成矩阵的函数,一种是eye(),一种是identity(),不同的是identity()只能生成方阵,即行数和列数相等,通过查看源码发现,identity()的返回值是eye(n,dtype=dtype),即调用了eye函数。
而eye提供了更多的参数,N:行数,M:列数,如果不传递,则和行数相同,K:0为主对角线,是默认值,正值为对角线上方,负值为对角线下方(个人理解为以主对角线为基准的偏移量),dtype:矩阵元素类型
import numpy as np
'''构造对角线为1,其他为0的5*5矩阵'''
a=np.eye(5,5,0,int)
'''构造根据主对角线向上偏移一位的5*5矩阵'''
b=np.eye(5,5,1,int)
'''构造根据主对角线向下偏移一位的5*5矩阵'''
c=np.eye(5,5,-1,int)
print(2*a+b+c)