前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >用Python的Numpy求解线性方程组

用Python的Numpy求解线性方程组

作者头像
拓端
发布2021-01-13 14:42:48
1.4K0
发布2021-01-13 14:42:48
举报
文章被收录于专栏:拓端tecdat拓端tecdat

原文链接:http://tecdat.cn/?p=8445

在本文中,您将看到如何使用Python的Numpy库解决线性方程组。

什么是线性方程组?

维基百科将线性方程组定义为:

在数学中,线性方程组(或线性系统)是两个或多个涉及同一组变量的线性方程的集合。

解决线性方程组的最终目标是找到未知变量的值。这是带有两个未知变量的线性方程组的示例:

等式1:

代码语言:javascript
复制
  1. 4x + 3y = 20
  2. -5x + 9y = 26

为了解决上述线性方程组,我们需要找到xy变量的值。解决方法有多种,例如消除变量,克莱默规则,矩阵解决方案。在本文中,我们将介绍矩阵解决方案。

在矩阵解中,要求解的线性方程组以矩阵形式表示AX = B。例如,我们可以用矩阵形式表示等式1,如下所示:

代码语言:javascript
复制
  1. A = [[ 4 3]
  2. [-5 9]]
  3. X = [[x]
  4. [y]]
  5. B = [[20]
  6. [26]]

要查找的值xy变量方程1,我们需要找到在矩阵中的值X。为此,我们可以采用矩阵逆的点积A和矩阵B,如下所示:

代码语言:javascript
复制

X = inverse(A).B

 
 

用numpy求解线性方程组

要求解线性方程组,我们需要执行两个操作:矩阵求逆和矩阵点积。Python的Numpy库支持这两种操作。如果尚未安装Numpy库,则可以使用以下pip命令:

代码语言:javascript
复制

$ pip install numpy

现在让我们看看如何使用Numpy库解决线性方程组。

使用inv()和dot()方法

首先,我们将找到A在上一节中定义的矩阵逆。

首先让我们A在Python中创建矩阵。要创建矩阵,array可以使用Numpy模块的方法。矩阵可以视为列表列表,其中每个列表代表一行。

在以下脚本中,我们创建一个名为的列表m_list,其中进一步包含两个列表:[4,3][-5,9]。这些列表是矩阵中的两行A。要A使用Numpy 创建矩阵,将m_list传递给array方法,如下所示:

代码语言:javascript
复制

import numpy as np
m_list = [[4, 3], [-5, 9]]A = np.array(m_list)

为了找到矩阵的逆,将矩阵传递给linalg.inv()Numpy模块:

代码语言:javascript
复制


inv_A = np.linalg.inv(A)
print(inv_A)

下一步是找出矩阵的逆矩阵之间的点积A和矩阵B。重要的是要提一下,只有在矩阵的维度相等的情况下,才可能在矩阵之间获得矩阵点积,即,左矩阵的列数必须与右矩阵的行数匹配。

要使用Numpy库查找点积,使用linalg.dot()函数。

代码语言:javascript
复制

B = np.array([20, 26])X = np.linalg.inv(A).dot(B)
print(X)

输出:

代码语言:javascript
复制
代码语言:javascript
复制
[2. 4.]

 

验证一下,如果在方程式中插入x4替换未知数,您将看到结果为20。

现在,让我们解决由三个线性方程组成的系统,如下所示:

代码语言:javascript
复制
  1. 4x + 3y + 2z = 25
  2. -2x + 2y + 3z = -10
  3. 3x -5y + 2z = -4

可以使用Numpy库按以下方式求解以上方程式:

公式2:

代码语言:javascript
复制



print(X)

在上面的脚本中,linalg.inv()linalg.dot()方法链接在一起。该变量X包含方程式2,并输出如下:

代码语言:javascript
复制
[ 5.  3. -2.]
代码语言:javascript
复制

未知数xy和的值分别是5、3 z和-2。您可以将这些值代入公式2并验证其正确性。

使用solve()方法

在前两个示例中,我们使用linalg.inv()linalg.dot()方法来找到方程组的解。但是,Numpy库包含该linalg.solve()方法,该方法可用于直接找到线性方程组的解:

代码语言:javascript
复制



print(X2)

输出:

代码语言:javascript
复制


[ 5.  3. -2.]

您可以看到输出与以前相同。

一个真实的例子

让我们看看如何使用线性方程组来解决实际问题。

假设有一个卖水果的人一天就卖出了20个芒果和10个橘子,总价为350元。第二天,他以500元的价格出售了17个芒果和22个橙子。如果这两天的水果价格都保持不变,那么一个芒果和一个橙子的价格是多少?

使用两个线性方程组可以轻松解决此问题。

假设一个芒果x的价格为,一个橙子的价格为y。上面的问题可以这样转换:

代码语言:javascript
复制
  1. 20x + 10y = 350
  2. 17x + 22y = 500

上面的方程组的解决方案如下所示:

代码语言:javascript
复制


X = np.linalg.solve(A,B)
print(X)

这是输出:

代码语言:javascript
复制


[10. 15.]

输出显示,一个芒果的价格为10元,一个橙子的价格为15元。

结论

本文介绍了如何使用Python的Numpy库解决线性方程组。您可以使用linalg.inv()linalg.dot()方法来求解线性方程组,也可以简单地使用solve()方法。solve()方法是首选方法。

本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2020-12-30,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 拓端数据部落 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 原文链接:http://tecdat.cn/?p=8445
  • 什么是线性方程组?
  • 用numpy求解线性方程组
  • 结论
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档