温馨提示×

怎么使用SciPy求解线性方程组

小亿
109
2024-05-24 11:34:13
栏目: 编程语言

SciPy库中提供了很多求解线性方程组的方法,其中最常用的是使用scipy.linalg.solve函数。下面是一个简单的例子来展示如何使用SciPy求解线性方程组。

假设我们有一个线性方程组:

3x + 2y - z = 1
2x - 2y + 4z = -2
-x + 0.5y - z = 0

我们可以将其表示成矩阵形式 Ax = b,其中

A = [[3, 2, -1],
     [2, -2, 4],
     [-1, 0.5, -1]]
b = [1, -2, 0]

接下来,我们可以使用scipy.linalg.solve函数来求解该线性方程组:

import numpy as np
from scipy.linalg import solve

A = np.array([[3, 2, -1], [2, -2, 4], [-1, 0.5, -1]])
b = np.array([1, -2, 0])

x = solve(A, b)

print(x)

运行以上代码,我们可以得到方程组的解为 x = [1, -2, -2]

0