Python PuLP - Unable to Model Non-Square Matrix

Question

I am having issues with setting up constraints using both input arrays from excel and variable arrays within PuLP.

It appears the model only works with square matrices and my final code has a matrix that is 365x24. The code below has a matrix of 5x6 and throws an error when running saying index out of range.

I have provided the code that I am working with so far.

Sheet11 Matrix

91  37  36  38  33  16
1   36  59  29  23  4
25  74  72  39  69  1
22  68  48  70  12  41
98  86  75  16  99  12

Sheet12 Matrix

59  63  66  57  4   15
26  33  75  71  21  2
37  88  89  1   90  3
91  48  27  24  23  14
68  13  61  37  77  20

from pulp import *
from pandas import *
import pandas as pd
import numpy as np
import xlrd

model = pulp.LpProblem("Basic Model", pulp.LpMinimize)

YPER = 5
HE = 6

yearlyhours = []
yearlyhours = [(i,j) for i in range(YPER) for j in range(HE)]

book = xlrd.open_workbook('Stack.xlsx')
sheet11 = book.sheet_by_name('Sheet11')
sheet12 = book.sheet_by_name('Sheet12')
sheet13 = book.sheet_by_name('Sheet13')

TEST = [[sheet11.cell_value(i,j) for i in range(YPER)] for j in range(HE)]
YAPR = [[sheet12.cell_value(i,j) for i in range(YPER)] for j in range(HE)]
MAPR = [[sheet13.cell_value(i,j) for i in range(YPER)] for j in range(HE)]

YAHL = pulp.LpVariable.dicts("YAHL", (range(YPER), range(HE)), lowBound=0, cat='Continuous')
MAHL = pulp.LpVariable.dicts("MAHL", (range(YPER), range(HE)), lowBound=0, cat='Continuous')

##OBJECTIVE##
model += pulp.lpSum([YAPR[i][j] * YAHL[i][j] + MAPR[i][j] * MAHL[i][j] for i in range(YPER) for j in range(HE)]), 'Sum_of_Value'

for i,j in yearlyhours:
    model += pulp.lpSum([YAHL[i][j] + MAHL[i][j]]) == ([TEST[i][j]])

LpSolverDefault.msg = 1
model.writeLP('Opt.lp')
model.solve()
pulp.LpStatus[model.status]
print("Status:", LpStatus[model.status])
obj = value(model.objective)
print("Total Cost: ${}".format(obj,2))
print('\n')

When changing to a non-square matrix the following error message is given:

IndexError: list index out of range

Any help appreciated as I am still learning.

Answer 1

I think that the issue is the following:

YAPR = [[sheet12.cell_value(i,j) for i in range(YPER)] for j in range(HE)]

Here you create a list of list (YAPR) to represent your matrix. The outer list has a length of HE and the inner lists have a length of YPER.

model += pulp.lpSum([YAPR[i][j] * YAHL[i][j] + MAPR[i][j] * MAHL[i][j] for i in range(YPER) for j in range(HE)]), 'Sum_of_Value'

Here you address the list of list YAPR as if the outer list had a size of YPER and the inner list a size of HE. This is causing your out of bounds exception. You do not have the issue with square matrix because then YPER=HE which means that swapping the dimensions does not cause to go out of bounds.

You probably want to review your complete model because I saw those indices inversion on multiple lines.