有没有更快的方法使用 pulp 为线性规划定义变量？

Question

我正在尝试建立一个涉及 3 个工厂和 4 个配送中心的线性规划问题。目标是最小化成本，但每个工厂配送中心组合都有不同的相关成本。

有没有更好的方法来使用 pulp 或其他库创建变量，这样我就不必写那么多了。目前我的是这样的。

from pulp import *  
#define the problem   
prob=LpProblem('transportation', LpMinimize)  
#create Variables  
plant1_center1 = LpVariable('plant1_center1', lowBound=0, cat='Integer')       
plant1_center2 = LpVariable('plant1_center2', lowBound=0, cat='Integer')  
plant1_center3 = LpVariable('plant1_center3', lowBound=0, cat='Integer')  
plant1_center4 = LpVariable('plant1_center4', lowBound=0, cat='Integer')  
plant2_center1 = LpVariable('plant2_center1', lowBound=0, cat='Integer')  
plant2_center2 = LpVariable('plant2_center2', lowBound=0, cat='Integer')   
plant2_center3 = LpVariable('plant2_center3', lowBound=0, cat='Integer')   
plant2_center4 = LpVariable('plant2_center4', lowBound=0, cat='Integer')   
plant3_center1 = LpVariable('plant3_center1', lowBound=0, cat='Integer')   
plant3_center2 = LpVariable('plant3_center2', lowBound=0, cat='Integer')   
plant3_center3 = LpVariable('plant3_center3', lowBound=0, cat='Integer')   
plant3_center4 = LpVariable('plant3_center4', lowBound=0, cat='Integer')

它有效，但我讨厌每次都必须创建这样的变量。

Answer 1

对行（植物）和列（分布中心）使用字符串插值和几个循环如何：

from pulp import *

def make_var(i, j):
    return LpVariable(f"plant{i}_center{j}", lowBound=0, cat="Integer")

n_plants = 3
n_dist_centers = 4
prob = LpProblem("transportation", LpMinimize)
plants = [
    [make_var(i, j) for j in range(n_dist_centers)] 
    for i in range(n_plants)
]

for row in plants:
    print(row)

我已经对其进行了零索引，因此您可以使用 plants[plant_number][dist_center_number] 访问特定的 plant，但是如果您需要使用 range(1, n_plants + 1)，您应该能够轻松地对其进行 1 索引].

请参阅 A Transportation Problem，其中提供了您可以尝试的类似模式。

有没有更快的方法使用 pulp 为线性规划定义变量？

Is there a faster way to define variables using pulp for linear programming?

python

variables

pulp