跳转到主内容
趣航编程网 - 趣学编程,启航技术之路!

PuLP 教程:如何约束每项任务仅由一名员工执行(使用二元变量与大M法)

本文详解如何在 pulp 中建模“每项任务最多分配给一名员工”的逻辑约束——通过引入二元分配变量和大m法将非线性条件(如 x > 0)转化为线性整数规划约束。 本文详解如何在 pulp 中建模“每项任务最多分配给一名员工”的逻辑约束——通过引入二元分配变量和大m法将非线性条件(如 x > 0)转化为线性整数规划约束。 在资源分配类优化问题中,一个常见但易被忽视的关键业务规则是: 某项具体任务(action)应由至多一名员工全权负责 ,而非多人分摊。这不同于简单的工时上限或工作量上限约束——它本质上是一个逻辑约束(“若某员工参与该任务,则其他人必须为0”),无法直接用 vars[action][m] > 0 这类比较表达,因为 PuLP 的 LpVariable 不支持与整数的布尔比较(会触发 TypeError: '>' not supported between instances of 'LpVariable' and 'int')。 解决这一问题的标准建模方法是 引入二元(binary)分配变量 + 大M(Big-M)线性化技巧 : ✅ assign[action][person] ∈ {0,1}:表示该员工是否被分配到此项任务(1 = 是,0 = 否); ✅ hours[action][person] ≥ 0:连续变量,表示该员工实际在此任务上投入的工时; ⚠️ 关键耦合约束:hours[action][person] ≤ assign[action][person] × M,其中 M 是足够大的常数(如 max(最大可用工时, 最大可用工作量))。该式确保:若 assign = 0,则 hours 必须为 0;若 assign = 1,则 hours 可自由取 [0, M] 内值(再由其他约束收紧)。 以下为完整、可运行的 PuLP 实现(已结构化并增强可读性):
import pandas as pd import pulp # 定义任务数据(索引为 action) actions = pd.DataFrame( index=pd.Index(name='action', data=['A', 'B', 'C', 'D']), data={ 'value': [5, 2, 1, 1], # 每单位工时的价值 'max_work': [8, 4, 12, 24], # 该任务最多可完成的工作量(工时上限) } ) # 定义员工数据(索引为 person) people = pd.DataFrame( index=pd.RangeIndex(name='person', start=1, stop=6), # 5名员工:1~5 data={ 'max_hours': [7, 7, 6, 5, 5], # 每人每日最大可用工时 } ) # 构建所有 (action, person) 组合的决策变量 DataFrame combos = pd.DataFrame({ 'assign': pulp.LpVariable.matrix( name='assign', indices=(actions.index, people.index), cat=pulp.LpBinary # 二元变量:是否分配 ), 'hours': pulp.LpVariable.matrix( name='hours', indices=(actions.index, people.index), cat=pulp.LpContinuous, lowBound=0 ) }).stack([0, 1]).to_frame() # 展平为 MultiIndex DataFrame # 衍生列:便于后续计算目标函数与约束 combos['value_per_hour'] = actions['value'].reindex(combos.index.get_level_values('action')).values combos['value'] = combos['hours'] * combos['value_per_hour'] # 按员工/任务聚合总工时(用于约束) people['total_hours'] = combos['hours'].groupby('person').sum() actions['total_work'] = combos['hours'].groupby('action').sum() # 创建优化问题 prob = pulp.LpProblem(name='PlanningActions', sense=pulp.LpMaximize) prob.setObjective(pulp.lpSum(combos['value'])) # 最大化总价值 # 【约束1】每人总工时 ≤ 其可用工时 for person in people.index: prob.addConstraint( name=f'hours_limit_{person}', constraint=people.loc[person, 'total_hours'] <= people.loc[person, 'max_hours'] ) # 【约束2】每项任务总工时 ≤ 其最大工作量 for action in actions.index: prob.addConstraint( name=f'work_limit_{action}', constraint=actions.loc[action, 'total_work'] <= actions.loc[action, 'max_work'] ) # 【约束3】每项任务最多分配给1名员工(核心逻辑约束) for action in actions.index: prob.addConstraint( name=f'assign_exclusive_{action}', constraint=pulp.lpSum(combos.loc[action, 'assign']) <= 1 ) # 【约束4】大M法:绑定 hours 与 assign(关键!) M = 2 * max(actions['max_work'].max(), people['max_hours'].max()) # 安全上界 for (action, person), row in combos.iterrows(): prob.addConstraint( name=f'link_assign_hours_{action}_{person}', constraint=row['hours'] <= row['assign'] * M ) # 求解 prob.solve(pulp.PULP_CBC_CMD(msg=False)) assert prob.status == pulp.LpStatusOptimal, "求解未收敛至最优解" # 输出结果(转换为数值) results = combos.applymap(pulp.value).round(1) print("=== 分配与工时详情 ===") print(results[['assign', 'hours', 'value']].sort_index()) print("\n=== 员工总工时 ===") print(people[['max_hours', 'total_hours']].applymap(pulp.value).round(1)) print("\n=== 任务完成量 ===") print(actions[['value', 'max_work', 'total_work']].applymap(pulp.value).round(1))
? 注意事项与最佳实践 : M 的选取至关重要 :过大(如 1e6)会导致数值不稳定或弱松弛;过小(小于任何可能的 hours)会错误剪枝可行解。推荐设为 2 × max(全局工时上限, 全局工作量上限); 无需额外约束“每人最多一项任务” :本例业务未要求,若需可添加 pulp.lpSum(combos.loc[:, person, 'assign']) <= 1; 变量命名清晰化 :使用 matrix() + stack() 比嵌套字典更利于 Pandas 索引与聚合,大幅提升可维护性; 验证结果合理性 :检查 assign 列是否确实满足“每 action 行 sum ≤ 1”,且 hours 非零行与 assign == 1 完全一致。 该建模范式是整数规划中处理“存在性→数量”逻辑的通用解法,适用于排班、指派、设施选址等广泛场景。掌握它,便能跨越 PuLP 中“不可比较变量”的表层限制,精准刻画真实业务规则。

相关文章