您好我正在尝试解决一个带有两个边约束的线性方程组,一个是成功实现的,结果总和为1,但我需要另一个,每个解都应该是非负的。谁知道如何添加此约束?感谢
import numpy as np
import numpy.linalg as LA
import scipy.optimize as optimize
A = np.array([[.5, .3, .2], [.4, 6, .3], [.2, .3, .5]])
b = np.array([0, 0, 0])
x = LA.solve(A, b)
def f(x):
y = np.dot(A, x) - b
return np.dot(y, y)
cons = ({'type': 'eq', 'fun': lambda x: x.sum() - 1},{'type': 'eq', 'fun': lambda x: x >= 0})
res = optimize.minimize(f, [0, 0, 0], method='SLSQP', constraints=cons,
options={'disp': False})
xbest = res['x']
print(xbest)
答案 0 :(得分:2)
我假设这是你想要解决的方程系统:
.5x1 + .3x2 + .2x3 = 0
.4x1 + 6x2 + .3x3 = 0
.2x1 + .3x2 + .5x3 = 0
x1 + x2 + x3 =1
x1, x2, x3 >=0
使用scipy.optimize.linprog
可以轻松解决此问题。在这里,由于您没有目标函数,目标函数的系数将为[0., 0., 0.]
。
from scipy.optimize import linprog
print(linprog(c=[0., 0., 0.],
A_eq=[[.5, .3, .2], [.4, 6, .3], [.2, .3, .5], [1., 1., 1.]],
b_eq=[0., 0., 0., 1.],
bounds=(0, None)))
这应该会为您提供问题的结果。但是,您的系统没有可行的解决方案。您可以在此处找到有关scipy.optimize.linprog
的更多信息:http://docs.scipy.org/doc/scipy-0.15.1/reference/generated/scipy.optimize.linprog.html