线性代数Python计算：解可逆系数矩阵线性方程组

numpy的linalg模块中solve函数解可逆系数矩阵的线性方程组。该函数的调用接口为
$$\text{solve(A, b)}$$
参数A表示系数矩阵$\mathbf{A}$，b表示常数矩阵$\mathbf{b}$。若A表示的系数矩阵可逆，返回方程组$\mathbf{Ax}=\mathbf{b}$的解，否则报错。
例1 下图中(a)表示一电路网络，每条线上标出的数值是电阻，E点接地。向X，Y，Z和U四点通入电流，强度均为$100$安培。求X，Y，Z，U四点处的电位。

解：设定进入X，Y，Z，U四点流向电路的电路为$I_1$，$I_2$，$I_3$，$I_4$，$I_5$，$I_6$，$I_7$，$I_8$（见图中(b)）。运用基尔霍夫定律知，
$$\{\begin{array}{l}{I}_1+I_2-I_3=100\\ I_3+I_4-I_5=100\\ I_5+I_6-I_7=100\\ I_7+I_8-I_1=100\end{array}.$$
设X，Y，Z，U四点处的电位分别为$V_X$，$V_Y$，$V_Z$，$V_U$。由欧姆定律知$I=\frac{V}{R}$，其中I，R，V分别表示电流，电阻，电位。于是，
$I_1=V_X-V_U$，$I_2=6V_X$，$I_3=2V_Y-2V_X$，$I_4=7V_Y$
$I_5=3V_Z-3V_Y$，$I_6=8V_Z$，$I_7=4V_U-4V_Z$，$I_8=5V_U$
代入电流方程组得
$$\{\begin{array}{l}9V_X-2V_Y-V_U=100\\ -2V_X+12V_Y-3V_Z=100\\ -3V_Y+15V_Z-4V_U=100\\ -V_X-4V_Z+10V_U=100\end{array}.$$
该方程组的系数矩阵为$\mathbf{A}=\left(\begin{array}{cccc}9& -2& 0& -1\\ -2& 12& -3& 0\\ 0& -3& 15& -4\\ -1& 0& -4& 10\end{array}\right)$，常数矩阵$\mathbf{b}=\left(\begin{array}{c}100\\ 100\\ 100\\ 100\end{array}\right)$。用下列代码计算该方程组的解。

import numpy as np                          #导入numpy
from fractions import Fraction as F         #导入Fraction
np.set_printoptions(formatter={'all':lambda x:
                               str(F(x).limit_denominator())})
A=np.array([[9,-2,0,-1],                    #系数矩阵A
            [-2,12,-3,0],
            [0,-3,15,-4],
            [-1,0,-4,10]])
b=np.array([100,100,100,100])               #常数矩阵b
X=np.linalg.solve(A, b)                     #解方程
print(X.reshape(4,1))                       #输出解

程序的第5~8行设置方程组的系数矩阵A，第9行设置常数矩阵b，第10行调用numpy.linalg的solve函数，传递参数A和b，计算方程组$\mathbf{Ax}=\mathbf{b}$的解X。运行程序，输出

[[210100/12907]
 [188400/12907]
 [183300/12907]
 [223400/12907]]

即即X，Y，Z，U四点处电位$V_X=\frac{210100}{12907}$，$V_Y=\frac{188400}{12907}$，$V_Z=\frac{183300}{12907}$，$V_U=\frac{223400}{12907}$。
写博不易，敬请支持：
如果阅读本文于您有所获，敬请点赞、评论、收藏，谢谢大家的支持！
代码诚可贵，原理价更高。若为AI学，读正版书好。