numpy 解一道简单数学题
题目
A group took a trip on a bus, at 3 per child and 3.20 per adult for a total of 118.40.
They took the train back at 3.50 per child and 3.60 per adult for a total of 135.20.
How many children, and how many adults?
求解过程
设有x1个children,x2个adults，线性方程组为：
\[3x_1+3.2x_2 = 118.4
\]
\[3.5x_1+3.6x_2 = 135.2
\]
矩阵形式表示为：
\[\begin{bmatrix}
3 & 3.2 \\
3.5 & 3.6
\end{bmatrix}
\begin{bmatrix}
x_1 \\
x_2 \\
\end{bmatrix}
=
\begin{bmatrix}
118.4 \\
135.2
\end{bmatrix}
\]
化为：
\[\begin{bmatrix}
x_1 \\
x_2
\end{bmatrix}
=
{
\begin{bmatrix}
3 & 3.2 \\
3.5 & 3.6
\end{bmatrix}
}^{-1}
\begin{bmatrix}
118.4 \\
135.2
\end{bmatrix}
\]
接下来就是使用numpy求解了
代码
import numpy as np
a = np.matrix('3. 3.2;3.5,3.6')
b = np.matrix('118.4;135.2')
inverse_a = np.linalg.inv(a)
print(inverse_a)
result = inverse_a * b
print(result)
结果
\[inverse\_a =
\begin{bmatrix}
-9 & 8 \\
8.75 & -7.5
\end{bmatrix}
\]
\[result =
\begin{bmatrix}
16 \\
22
\end{bmatrix}
\]
也就是说是16个children，22个adults
