Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

How to define JuMP variables using for loop in Julia?

I am new to Julia, and I am trying to define an optimization problem with JuMP. I have a lot of variables (x1,x2,x3....) that I am trying to define using a for loop. I want to have the code:

@variable(m, x1>=0)
@variable(m, x2>=0) ... 

However I wanted to use a for loop so I did not have to define every variable manually.
Here is what I have so far:

m = Model()
for i = 1:2
    @variable(m,string('x',i)>=0)
end 

I know the string('x',i) part is not right but I am not sure how to do this using Julia.

like image 779
Cam Avatar asked Aug 04 '15 14:08

Cam


2 Answers

It looks like you want an array of x variables.

From the JuMP docs, you can make an array by using array syntax in your definition.

@variable(m, x[1:2] >= 0)
@variable(m, y[1:M,1:N] >= 0)
like image 161
Andrew Avatar answered Oct 13 '22 00:10

Andrew


You can add indices to your variables using @variable. The following are all valid in JuMP:

m = Model()
@variable(m, x[1:2] >= 0)
@variable(m, boringvariable[1:9,1:9,1:9])
@variable(m, 0 <= pixel_intensity[1:255,1:255] <= 1)
@variable(m, bit_pattern[0:8:63], Bin)
N = 5, M = 10
@variable(m, trucks_dispatched[i=1:N,j=1:M] >= 0, Int)
items = [:sock,:sandal,:boot]
max_stock = [:sock => 10, :sandal => 13, :boot => 5]
@variable(m, 0 <= stock_levels[item=items] <= max_stock[item])
like image 35
IainDunning Avatar answered Oct 12 '22 23:10

IainDunning