如何从PYOMO索引变量输出中提取数据生成面-材料匹配字典

实现方案 #

你可以直接利用Python内置max()函数的key参数指定比较规则，无需单独先取最大值再反向匹配材料，代码如下：

import pyomo.environ as pyo

# 已有定义的参数
materials = ['steel', 'alum', 'copper', 'zinc']
# 定义所有待匹配的面
faces = [f'Face {i}' for i in range(1, 7)]
# 过滤求解器浮点残差的阈值，可根据需求调整
EPS = 1e-6

# 字典推导式一行生成结果
selected_materials = {
    face: max(
        materials, 
        key=lambda m: pyo.value(mdl.x[m, face])
    ) 
    for face in faces
}

# 可选：增加数值校验，过滤极小残差导致的错误匹配
for face, mat in selected_materials.items():
    if pyo.value(mdl.x[mat, face]) < EPS:
        # 可根据业务需求替换为默认值或抛出异常
        selected_materials[face] = None

代码说明 #

• max()函数传入materials列表作为待选对象，通过key参数指定比较规则为对应材料的x变量取值，会直接返回取值最大的材料名称，无需额外拆分元组。
• 数值稳定性处理：你给出的变量输出中存在大量1e-8量级的极小值，属于混合整数求解器的正常浮点残差，对应二进制变量的0值，增加阈值校验可以避免这类极小值偶然超过0导致的匹配错误。
• 如果存在多个材料x值相等的场景，max()会返回materials列表中排序更靠前的材料，如需处理并列最大值的情况可自行扩展判断逻辑。

运行上述代码得到的selected_materials就和你期望的输出格式完全一致。

内容的提问来源于stack exchange，提问作者Wilson Mendes