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

转载于:https://www.cnblogs.com/WeyneChen/p/6694132.html