Python约束线性优化

时间:2016-06-14 20:00:06

标签: python scipy

您好我正在尝试解决一个带有两个边约束的线性方程组,一个是成功实现的,结果总和为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)

1 个答案:

答案 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