PyPSA设置Gurobi时间限制方法

本文档旨在指导用户如何在PyPSA模型中使用Gurobi求解器时设置时间限制。我们将详细介绍如何配置solver_options参数，以及如何处理求解器因达到时间限制而中断的情况，确保模型在可控的时间范围内求解，并获得可用的次优解。

在PyPSA模型中配置Gurobi求解器的时间限制

在使用PyPSA构建电力系统优化模型时，经常需要使用优化求解器来找到最优解。Gurobi是一个常用的商业优化求解器，可以通过PyPSA进行调用。为了控制求解时间，可以为Gurobi求解器设置时间限制。

以下是如何在PyPSA模型中设置Gurobi求解器时间限制的步骤：

1. 定义求解器选项 (solver_options):

   在调用network.lopf()或network.optimize()函数时，通过solver_options参数传递Gurobi的配置选项。TimeLimit参数用于设置求解器的最大运行时间，单位为秒。

   solver_options = {
       'LogFile': "gurobiLog", # 可选：指定日志文件
       'MIPGap': 0.001,        # 可选：设置MIP相对间隙
       'BarConvTol': 0.01,     # 可选：设置 Barrier 收敛容差
       'TimeLimit': 200,       # 设置时间限制为200秒
   }
   
   # 使用 network.optimize() 代替 network.lopf()，后者已被弃用
   network.optimize(network.snapshots, solver_name='gurobi', solver_options=solver_options)

   注意: network.lopf() 方法已经过时，建议使用 network.optimize() 方法。

2. 检查Gurobi日志文件 (可选):

   通过设置LogFile参数，可以查看Gurobi的日志文件，确认时间限制是否生效，以及求解器的运行状态。

   Gurobi 10.0.1 (win64) logging started Tue Dec 12 19:51:07 2023
   
   Set parameter LogFile to value "gurobiLog"
   Set parameter MIPGap to value 0.001
   Set parameter BarConvTol to value 0.01
   Set parameter TimeLimit to value 5

   在日志文件中，可以观察到TimeLimit参数被正确设置。

3. 处理求解器中断:

   当Gurobi求解器达到时间限制时，会中断求解过程。默认情况下，PyPSA会抛出一个ValueError异常，提示求解器状态为"aborted"。为了避免程序崩溃，需要处理这个异常，并尝试获取已找到的次优解。

   原因分析:

   出现 ValueError: Cannot load a SolverResults object with bad status: aborted 错误的原因是，Gurobi 在达到时间限制后中断，导致 PyPSA 无法加载完整的求解结果。

   解决方法:

   在 PyPSA 中，即使求解器由于时间限制而停止，通常仍然可以访问到求解器找到的最佳可行解。你需要确保你的代码能够处理求解器非正常终止的情况，并提取可用的结果。

   以下是一个处理求解器中断的示例代码：

   try:
       network.optimize(network.snapshots, solver_name='gurobi', solver_options=solver_options)
   except ValueError as e:
       print(f"求解器由于以下原因中断: {e}")
       print("尝试提取已找到的最佳解...")
       # 在这里添加提取和使用次优解的代码
       # 例如，检查 network.results 对象是否包含有效结果
       if hasattr(network, 'results') and network.results is not None:
           print("成功提取部分解。")
           # 可以进一步处理 network.results 中的数据
       else:
           print("未能提取任何解。")

   注意: 即使设置了时间限制，Gurobi 也可能在达到最优解之前停止。因此，需要根据实际情况设置合适的MIPGap参数，以便在可接受的时间范围内找到足够好的解。

完整示例代码

以下是一个完整的示例代码，展示了如何在PyPSA模型中使用Gurobi求解器设置时间限制，并处理求解器中断的情况：

import pypsa
import numpy as np
import pandas as pd

# 创建一个 PyPSA 网络
network = pypsa.Network()

# 添加一个负载总线
network.add("Bus", "Bus")

# 定义时间快照
snapshots = pd.date_range("2022-01-01", "2022-01-02", freq="H")
network.set_snapshots(snapshots)

# 添加负载
load_profile = np.random.randint(2800, 3300, len(snapshots))
network.add("Load", "Load profile", bus="Bus", p_set=load_profile)

# 定义发电机数据
generator_data = {
    'coal1': {'capacity': 800, 'carrier': 'Coal', 'variable cost': 10, 'co2_emission_factor': 0.95},
    'gas1': {'capacity': 600, 'carrier': 'Gas', 'variable cost': 12, 'co2_emission_factor': 0.45},
    'solar1': {'capacity': 150, 'carrier': 'Solar', 'variable cost': 1, 'co2_emission_factor': 0.0},
}

# 添加发电机
for name, data in generator_data.items():
    network.add("Generator", name,
                bus="Bus",
                carrier=data['carrier'],
                p_nom=data['capacity'],
                marginal_cost=data['variable cost'],
                )

# 添加载波
network.add("Carrier", "Coal", co2_emissions=0.95)
network.add("Carrier", "Gas", co2_emissions=0.45)
network.add("Carrier", "Solar", co2_emissions=0)

# 添加全局约束 (CO2 排放限制)
network.add(
    "GlobalConstraint",
    "CO2Limit",
    carrier_attribute="co2_emissions",
    sense="<=",
    constant=50000000,
)

# 定义求解器选项，设置时间限制为 5 秒
solver_name = "gurobi"
solver_options = {
    'LogFile': "gurobiLog",
    'MIPGap': 0.001,
    'BarConvTol': 0.01,
    'TimeLimit': 5,
}

# 尝试优化模型，并处理可能的异常
try:
    network.optimize(network.snapshots, solver_name=solver_name, solver_options=solver_options)
except ValueError as e:
    print(f"求解器由于以下原因中断: {e}")
    print("尝试提取已找到的最佳解...")
    if hasattr(network, 'results') and network.results is not None:
        print("成功提取部分解。")
        # 在这里可以进一步处理 network.results 中的数据
        # 例如，可以访问 network.generators_t.p 来获取发电机的出力
        dispatch = network.generators_t.p
        print(dispatch)
    else:
        print("未能提取任何解。")

# 导出结果到 CSV 文件 (可选)
csv_folder_name = 'model_dump'
network.export_to_csv_folder(csv_folder_name)

总结

通过本文档，您学习了如何在PyPSA模型中使用Gurobi求解器设置时间限制，以及如何处理求解器因达到时间限制而中断的情况。掌握这些技巧可以帮助您在实际应用中更好地控制求解时间，并获得可用的次优解。记住使用 network.optimize() 替代已经过时的 network.lopf() 方法。