利用python numpy实现矩阵的简单变换

  想生成如下的矩阵，主对角线是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)