跳转到主内容
新派编程网 - 新潮编程理念,全新技术课堂!

PuLP 教程:如何约束每个任务仅由一名员工执行

本文详解在 pulp 中建模“单任务单人”约束的方法——通过引入二元分配变量与连续工时变量,并结合大m法(big-m)建立逻辑关联,确保每个任务最多被一名员工承担。 本文详解在 pulp 中建模“单任务单人”约束的方法——通过引入二元分配变量与连续工时变量,并结合大m法(big-m)建立逻辑关联,确保每个任务最多被一名员工承担。 在资源分配类整数规划问题中,一个常见但易被忽视的关键约束是: 每个任务(action)只能由至多一名员工执行 。初学者常试图直接对连续型决策变量(如 vars[action][person] > 0)做逻辑判断,但这在 PuLP 中不可行——因为 LpVariable 对象不支持 Python 原生比较运算符(如 >),会抛出 TypeError。 正确解法是采用 混合整数规划(MIP)的标准建模技巧:分离“是否分配”与“分配多少”两个维度 : 定义二元变量 assign[action, person] ∈ {0, 1} :表示该员工是否被指派到该任务; 定义连续变量 hours[action, person] ≥ 0 :表示该员工在该任务上实际投入的工时; 用大M约束将二者逻辑绑定 :hours[action, person] ≤ assign[action, person] × M,其中 M 是足够大的上界(例如所有 max_work 和 max_hours 的最大值的两倍),确保当 assign = 0 时,hours 必须为 0;当 assign = 1 时,hours 可自由取 [0, M] 内值(再由其他约束收紧)。 以下是完整、可运行的建模示例(已适配原问题数据结构):
import pandas as pd import pulp # 定义任务数据 actions = pd.DataFrame( index=pd.Index(name='action', data=['ActionA', 'ActionB', 'ActionC', 'ActionD']), data={ 'value': [5, 2, 1, 1], 'max_work': [8, 4, 12, 24], # 每个任务最多可完成的工作量 } ) # 定义员工数据 people = pd.DataFrame( index=pd.RangeIndex(name='person', start=1, stop=6), # ID: 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() # 层级索引:(action, person) # 添加辅助列:单位工时价值 & 总价值 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('PlanningActions', pulp.LpMaximize) prob.setObjective(pulp.lpSum(combos['value'])) # 【核心约束1】每个任务最多分配给1名员工 for action, group in combos['assign'].groupby('action'): prob.addConstraint( pulp.lpSum(group) <= 1, name=f'excl_{action}' ) # 【核心约束2】每个员工总工时不超过其上限 for person, group in combos['hours'].groupby('person'): prob.addConstraint( pulp.lpSum(group) <= people.loc[person, 'max_hours'], name=f'hours_{person}' ) # 【核心约束3】每个任务总工作量不超过其上限 for action, group in combos['hours'].groupby('action'): prob.addConstraint( pulp.lpSum(group) <= actions.loc[action, 'max_work'], name=f'work_{action}' ) # 【关键逻辑约束】工时必须以分配为前提(Big-M) M = 2 * max(actions['max_work'].max(), people['max_hours'].max()) for (action, person), row in combos.iterrows(): prob.addConstraint( row['hours'] <= row['assign'] * M, name=f'link_{action}_{person}' ) # 求解 prob.solve(pulp.PULP_CBC_CMD(msg=False)) assert prob.status == pulp.LpStatusOptimal, "求解未收敛" # 输出结果 results = combos.map(pulp.value) print("分配与工时详情:") print(results.round(1)) print("\n员工总工时使用情况:") print(people['total_hours'].map(pulp.value).round(1)) print("\n任务完成总量:") print(actions['total_work'].map(pulp.value).round(1))
✅ 关键注意事项 : 勿混淆 LpVariable.dicts() 与 LpVariable.matrix() :后者更利于 Pandas 向量化操作,避免嵌套循环,代码更清晰、易维护; 大M值需合理选取 :过大可能导致数值不稳定或弱松弛,过小则破坏可行性;推荐取相关参数上界的 1.5–2 倍; 二元变量命名语义明确 :如 assign 比 x 更具可读性,便于后期调试与协作; 约束命名规范 :使用 name= 参数为每个约束指定唯一标识,print(prob) 时可直观查看模型结构。 该建模范式不仅适用于“单任务单人”,还可扩展至“每任务至多 K 人”、带技能匹配、优先级调度等复杂场景,是运筹优化工程实践中的基础且强大工具。

相关文章