(使用numpy)如果我们重写线性方程组
x - y + 2z = 5
y - z = -1
z = 3
作为矩阵方程
A x = b
与
A = np.array([[ 1, -1, 2],
[ 0, 1, -1],
[ 0, 0, 1]])
和
b = np.array([5, -1, 3])
然后x
可以使用找到np.linalg.solve
:
import numpy as np
A = np.array([(1, -1, 2), (0, 1, -1), (0, 0, 1)])
b = np.array([5, -1, 3])
x = np.linalg.solve(A, b)
产量
print(x)
# [ 1. 2. 3.]
我们可以检查一下A x = b
:
print(np.dot(A,x))
# [ 5. -1. 3.]
assert np.allclose(np.dot(A,x), b)