求解!!如何用Python写出这样的矩阵

img

求最简单的方法!

img

我是写了这样三个矩阵,但我不知道怎么把它们合起来。。
求各路大/佬帮帮忙,作/业完成不了啊。。

你好,熟练使用diag就可以达到目的

import numpy as np
a = np.diag(4*np.ones(8))+np.diag(np.ones(7),-1)+np.diag(np.ones(7),1);
a[0][0]=2
a[-1][-1]=2
print(a)

效果:

[[2. 1. 0. 0. 0. 0. 0. 0.]
 [1. 4. 1. 0. 0. 0. 0. 0.]
 [0. 1. 4. 1. 0. 0. 0. 0.]
 [0. 0. 1. 4. 1. 0. 0. 0.]
 [0. 0. 0. 1. 4. 1. 0. 0.]
 [0. 0. 0. 0. 1. 4. 1. 0.]
 [0. 0. 0. 0. 0. 1. 4. 1.]
 [0. 0. 0. 0. 0. 0. 1. 2.]]