2. 线性规划:两阶段法python代码

1. 补充问题

上一节中的代码在运行时还有很多细节没有处理,这里补充两个比较重要的情况:

  1. 存在等式约束
    如果有等式约束,那么就没法通过添加松弛变量直接给出初始可行解,需要用大M法或者两阶段法求初始可行解。计算机求解一般使用二阶段法,即首先给等式约束条件添加人工变量,使得问题有一个初始可行解。注意人工变量和松弛变量/剩余变量不一样。松弛变量/剩余变量是用于将不等式变为等式(标准形),而人工变量则是在等式约束中额外添加一个变量,这个变量最后一定要等于0才行,否则就是没有初始可行解,不能进入第二阶段。
    在第一阶段,每个不等式约束对应一个松弛变量,每个等式约束对应一个人工变量。另外,b<0的不等式约束也需要添加人工变量,因为对应的松弛变量等于b不能作为基变量。
  2. 没有有限最优解(目标函数无界)
    对应的情况是:目标函数中有某个系数大于0,对应变量在约束条件中的系数全都≤0,问题结束,没有最优解。

2. python算法实现

首先假设问题除了不等式约束,还有等式约束:

求解问题为:min c*x
s.t.A_ub * x ≤ b_ub
    A_eq * x = b_eq

定义maxiter为最大迭代次数,tol为求解精度,OptimizeResult为格式输出函数,_solve_simplex为上一节的求解函数。完整的求解步骤如下:

def linprog(c, A_ub=None, b_ub=None, A_eq=None, b_eq=None, maxiter=1000, tol=1.0E-12,):
    cc = np.asarray(c)
    f0 = 0
    n = len(c)
    mub = len(b_ub)
    meq = len(b_eq)
    m = mub+meq
    n_slack = mub
    n_artificial = meq + np.count_nonzero(bub < 0)
    Aub_rows, Aub_cols = A_ub.shape
    Aeq_rows, Aeq_cols = A_eq.shape
    slcount = 0
    avcount = 0
    status = 0
    messages = {0: "Optimization terminated successfully.",
                1: "Iteration limit reached.",
                2: "Optimization failed. Unable to find a feasible"
                   " starting point.",
                3: "Optimization failed. The problem appears to be unbounded.",
                4: "Optimization failed. Singular matrix encountered."}
    # 建立第一阶段单纯型表。
    T = np.zeros([m+2, n+n_slack+n_artificial+1])
    T[-2, :n] = cc
    T[-2, -1] = f0
    b = T[:-2, -1]
    if meq > 0:
        T[:meq, :n] = Aeq
        b[:meq] = beq
    if mub > 0:
        T[meq:meq+mub, :n] = Aub
        b[meq:meq+mub] = bub
        np.fill_diagonal(T[meq:m, n:n+n_slack], 1)

    #第一阶段的基向量basis,记录下标。
    basis = np.zeros(m, dtype=int)
    r_artificial = np.zeros(n_artificial, dtype=int)
    for i in range(m):
        if i < meq or b[i] < 0:
            basis[i] = n+n_slack+avcount
            r_artificial[avcount] = i
            avcount += 1
            if b[i] < 0:
                b[i] *= -1
                T[i, :-1] *= -1
            T[i, basis[i]] = 1
            T[-1, basis[i]] = 1
        else:
            basis[i] = n+slcount
            slcount += 1

    for r in r_artificial:
        T[-1, :] = T[-1, :] - T[r, :]
  
    #第一阶段求解
    nit1, status = _solve_simplex(T, n, basis)

    # 如果第一阶段的目标函数为0,则删除人工变量,进入第二阶段
    if abs(T[-1, -1]) <= sol:
        T = T[:-1, :]
        T = np.delete(T, np.s_[n+n_slack:n+n_slack+n_artificial], 1)
    else:
        status = 2

    if status != 0:
        message = messages[status]
        if disp:
            print(message)
        return OptimizeResult(x=np.nan, fun=-T[-1, -1], nit=nit1, status=status, message=message, success=False)
    nit2, status = _solve_simplex(T, n, basis, maxiter=maxiter-nit1, phase=2, callback=callback, tol=tol, nit0=nit1)

    solution = np.zeros(n+n_slack+n_artificial)
    solution[basis[:m]] = T[:m, -1]
    x = solution[:n]
    slack = solution[n:n+n_slack]
    obj = -T[-1, -1]
    return OptimizeResult(x=x, fun=obj, nit=int(nit2), status=status, slack=slack, message=messages[status], success=(status == 0))

下面是例子:

import numpy as np
c=np.array([2,3,-5])
A_ub=np.array([[-2,5,-1],[1,3,1]])
B_ub=np.array([-10,12])
A_eq=np.array([[1,1,1]])
B_eq=np.array([7])
res=linprog(-c,A_ub,B_ub,A_eq,B_eq)
print(res)

输出为

     fun: -14.571428571428571
 message: 'Optimization terminated successfully.'
     nit: 2
   slack: array([3.85714286, 0.57142857, 6.42857143, 7.        , 0.        ])
  status: 0
 success: True
       x: array([6.42857143, 0.57142857, 0.        ])
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 217,657评论 6 505
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,889评论 3 394
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 164,057评论 0 354
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,509评论 1 293
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,562评论 6 392
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,443评论 1 302
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,251评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,129评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,561评论 1 314
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,779评论 3 335
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,902评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,621评论 5 345
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,220评论 3 328
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,838评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,971评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,025评论 2 370
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,843评论 2 354

推荐阅读更多精彩内容

  • 本章涉及知识点1、线性规划的定义2、可行区域、目标函数、可行解和最优解3、转线性规划为标准型4、转线性规划为松弛型...
    PrivateEye_zzy阅读 38,533评论 2 36
  • 转载请注明出处 http://www.jianshu.com/p/dd0761a2fdfd作者:@贰拾贰画生 单纯...
    贰拾贰画生阅读 50,773评论 10 38
  • ​一般来说凸优化(Convex Optimization, CO)中最一般的是锥规划 (Cone Programm...
    史春奇阅读 5,074评论 1 6
  • 今天在家整理全域旅游的讲课资料。 全域旅游,概念没记住,记住了某省副省长的八个字,处处,时时,行行,人人。处处是景...
    西瓜猫猫阅读 171评论 0 0
  • 一、什么是Handler Handler是android给我们提供用来更新UI的机制,同时也是消息处理机制,可以通...
    瓦西大人阅读 216评论 1 3