Линейное программирование со Scipy

1

Я пытаюсь решить проблему линейного программирования со Scipy, я получаю ошибку, говоря, что размеры аргументов не совпадают. Но, похоже, это так, код и сообщение об ошибке ниже

Код

import numpy as np
from scipy import optimize as opt


k = 6
n = 3
indexes = [1, 2, 5, 6, 7, 9]
V = np.zeros((1, k))
count = 0
for ID in xrange(1, 4):
    ind = count * n + ID
    p = indexes.index(ind)
    V[0, p] = 1
    count += 1

bounds = []
for i in xrange(6):
    bounds.append((0, 1))
bounds = tuple(bounds)
W1 = np.zeros((3, 6))
W1[1, 2] = 0.4
W1[2, 3] = 0.5
b1 = np.transpose(np.zeros(3))
b1[1] = 0.8
b1[2] = 0.25

W3 = np.zeros((3, 6))
W3[1, 2] = 0.7
W3[2, 3] = 0.8
b3 = np.transpose(np.zeros(3))
b3[1] = 0.6
b3[2] = 0.2

EQ = np.vstack([W1, W3]).T
Eb = np.vstack([b1, b3]).T

print EQ.shape, "shape of A_eq"
print V.shape, "shape of c"

res = opt.linprog(c=V, A_eq=EQ, b_eq=Eb, bounds=bounds, options={"disp": True})

Сообщение об ошибке

ValueError: Invalid input for linprog with method = 'simplex'.  Number of columns in A_eq must be equal to the size of c
Теги:
scipy
linear-programming
arguments

1 ответ

1
Лучший ответ

Просто замените

res = opt.linprog(c=V, A_eq=EQ, b_eq=Eb, bounds=bounds, options={"disp": True})

от

res = opt.linprog(c=V[0], A_eq=EQ, b_eq=Eb, bounds=bounds, options={"disp": True})

Если вы печатаете V, вы увидите, что это список списка. Таким образом, данные, которые вы хотите, расположены под V[0]. Хотя оптимизация терпит неудачу.

Другой способ - переопределить ваш V как

V = np.zeros(k)

и затем использовать в цикле for

V[p] = 1.

Таким образом, вы можете придерживаться c=V в части оптимизации.

Ещё вопросы

Сообщество Overcoder
Наверх
Меню