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

Question

求最简单的方法！

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

Answer 1

你好，熟练使用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.]]