我目前需要一个类,它必须能够显示和解决像这样的方程式系统:
| 2x-4y+4z=8 |
| 34x+3y-z=30 |
| x+y+z=108 |
我认为编写一个类来将eqation系统的左侧事物转换为类似矩阵的对象是个好主意,这里是这个系统的自制矩阵:
/2 -4 4\
|34 3 -1|
\1 1 1/
我目前写了这个:
class mymatrix(object):
def __init__(self):
o11 = None
o12 = None
o12 = None
o21 = None
o22 = None
o23 = None
o31 = None
o32 = None
o33 = None
def set(row, column, value):
string = 'o'+str(row)+str(column)+' = '+str(value)
exec(string)
def solve(self, listwithrightsidethings):
#Here I want to solve the system. This code should read the three
#values out of the list and solves the system It should return the
#values for x, y and z in a tuple: (x, y, z)
pass
我搜索了一个模块来解决线性代数的问题,我找到了numpy。我在手册中搜索过,但没有找到我的问题解决方案
如何撰写solve
functoin?
编辑:
python应该像这样解释它
/o11, o21, o31\ 123
|o21, o22, o32| = 456
\o31, o32, o33/ 789
编辑:我想解决它3个vars,并将其作为元组
返回答案 0 :(得分:9)
您可以使用numpy.linalg.solve
:
import numpy as np
a = np.array([[2, -4, 4], [34, 3, -1], [1, 1, 1]])
b = np.array([8, 30, 108])
x = np.linalg.solve(a, b)
print x # [ -2.17647059 53.54411765 56.63235294]
答案 1 :(得分:5)
import numpy as np
a = np.array([[2, -4, 4], [34, 3, -1], [1, 1, 1]])
b = np.array([8, 30, 108])
try:
x = np.linalg.solve(a, b)
except LinAlgError:
x = np.linalg.lstsq(a, b)[0]