本文详解如何在 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 实现(已结构化并增强可读性):
?
注意事项与最佳实践
:
M 的选取至关重要
:过大(如 1e6)会导致数值不稳定或弱松弛;过小(小于任何可能的 hours)会错误剪枝可行解。推荐设为 2 × max(全局工时上限, 全局工作量上限);
无需额外约束“每人最多一项任务”
:本例业务未要求,若需可添加 pulp.lpSum(combos.loc[:, person, 'assign']) <= 1;
变量命名清晰化
:使用 matrix() + stack() 比嵌套字典更利于 Pandas 索引与聚合,大幅提升可维护性;
验证结果合理性
:检查 assign 列是否确实满足“每 action 行 sum ≤ 1”,且 hours 非零行与 assign == 1 完全一致。
该建模范式是整数规划中处理“存在性→数量”逻辑的通用解法,适用于排班、指派、设施选址等广泛场景。掌握它,便能跨越 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))