too many indices for array: array is 1-dimensional, but 2 were indexedが解決できません…

Question

最小化問題

pyomoでコスト最小化問題を計算しようとしていました。
線形の問題に対して2つの制約条件を設けて計算しようとしましたが、以下のようなエラーが出てしまいました。

発生している問題・エラーメッセージ

IndexError: too many indices for array: array is 1-dimensional, but 2 were indexed

該当のソースコード

import pyomo.environ as pyo
from pyomo.opt import SolverFactory
import numpy as np

# 変数・パラメータの設定

#係数行列の次元
N = 6
M = 2

#係数ベクトル、係数行列の作成
b = np.array([24261,132454,18026,40737,27865,2732], dtype = "float")
A = np.array([[0.43860,0.29240,0.27911,1,0.27911,0],[1.34575,0.82225,-0.21513,1.18625,1.56975,-1]], dtype = "float")
B = np.array([0,0,0,0,0,1], dtype = "float")
c = np.array([510183,0], dtype = "float")
d = np.array([100000])

#決定変数の設定
x = np.zeros(shape=(N), dtype = "float")

#決定変数の初期化用関数
xx = {i:x[i] for i in range(len(x))}
def init_x(model,i):
    return xx[i]

N_index_list = list(range(N))
M_index_list = list(range(M))

#不等式制約
def const(model,j):
    tmp = sum(A[j,i]*model.x[i] for i in N_index_list)
    return tmp == c[j]

def const2(model,j):
    tmp2 = sum(B[j,i]*model.x[i]for i in N_index_list)
    return tmp2 <= d[j]

#目的関数
def func(model):
    return sum(b[i] * model.x[i] for i in N_index_list)

#モデル
model = pyo.AbstractModel("LP-sample")

#決定変数をモデルに組み込む
model.x = pyo.Var(N_index_list, domain=pyo.NonNegativeReals, initialize = init_x)

#目的関数をモデルに組み込む
model.value = pyo.Objective(rule = func, sense = pyo.minimize)

#制約をモデルに組み込む
model.const = pyo.Constraint(M_index_list, rule = const)
model.const2 = pyo.Constraint(M_index_list, rule = const2)

#ソルバーの選択とオプション
solver_name = "glpk"
opt = SolverFactory(solver_name)

instance = model.create_instance()
instance.dual = pyo.Suffix(direction=pyo.Suffix.IMPORT) #双対変数取得用に必要
results = opt.solve(instance, tee = True) #tee=Trueでログ出力
instance.load(results)
results.write()

試したこと

補足情報（FW/ツールのバージョンなど）

制約条件を複数設定しましたが、これで本当にいいのかすらわかっていない状態です。
これを参考にしてプログラムは組んでいます。
https://takala.tokyo/takala_wp/2019/02/20/844/

Comments

[meg_]

エラーが発生した箇所はどこでしょうか？

[kk_nikmann]

このようなエラーが出ておりました

ERROR: Rule failed when generating expression for constraint const2 with index
0: IndexError: too many indices for array: array is 1-dimensional, but 2
were indexed
ERROR: Constructing component 'const2' from data=None failed:
IndexError: too many indices for array: array is 1-dimensional, but 2
were indexed
---------------------------------------------------------------------------
IndexError Traceback (most recent call last)
<ipython-input-69-8e225d568fd2> in <module>
----> 1 instance = model.create_instance()
2 instance.dual = pyo.Suffix(direction=pyo.Suffix.IMPORT) #双対変数取得用に必要
3 results = opt.solve(instance, tee = True) #tee=Trueでログ出力
4 instance.load(results)
5 results.write()

~\anaconda3\lib\site-packages\pyomo\core\base\PyomoModel.py in create_instance(self, filename, data, name, namespace, namespaces, profile_memory, report_timing, **kwds)
720 _namespaces.append(None)
721
--> 722 instance.load( data,
723 namespaces=_namespaces,
724 profile_memory=profile_memory )

~\anaconda3\lib\site-packages\pyomo\core\base\PyomoModel.py in load(self, arg, namespaces, profile_memory, report_timing)
777 msg = "Cannot load model model data from with object of type '%s'"
778 raise ValueError(msg % str( type(arg) ))
--> 779 self._load_model_data(dp,
780 namespaces,
781 profile_memory=profile_memory)

~\anaconda3\lib\site-packages\pyomo\core\base\PyomoModel.py in _load_model_data(self, modeldata, namespaces, **kwds)
830 continue
831
--> 832 self._initialize_component(modeldata, namespaces, component_name, profile_memory)
833 if False:
834 total_time = time.time() - start_time

~\anaconda3\lib\site-packages\pyomo\core\base\PyomoModel.py in _initialize_component(self, modeldata, namespaces, component_name, profile_memory)
881 declaration.name, _blockName, str(data) )
882 try:
--> 883 declaration.construct(data)
884 except:
885 err = sys.exc_info()[1]

~\anaconda3\lib\site-packages\pyomo\core\base\constraint.py in construct(self, data)
833 for index in self.index_set():
834 self._setitem_when_not_present(
--> 835 index, self.rule(block, index)
836 )
837 except Exception:

~\anaconda3\lib\site-packages\pyomo\core\base\util.py in __call__(self, parent, idx)
302 return self._fcn(parent, *idx)
303 else:
--> 304 return self._fcn(parent, idx)
305
306

<ipython-input-66-d454e3429ed2> in const2(model, j)
5
6 def const2(model,j):
----> 7 tmp2 = sum(B[j,i]*model.x[i]for i in N_index_list)
8 return tmp2 <= d[j]
9

<ipython-input-66-d454e3429ed2> in <genexpr>(.0)
5
6 def const2(model,j):
----> 7 tmp2 = sum(B[j,i]*model.x[i]for i in N_index_list)
8 return tmp2 <= d[j]
9

IndexError: too many indices for array: array is 1-dimensional, but 2 were indexed

Answer 1

エラーとしては、1次元の配列に、2つのインデックスが指定されているというような内容です。

どこでエラーが発生しているか書いてないですが、const2()の部分がおかしいです。

B = np.array([0,0,0,0,0,1], dtype = "float")

def const2(model,j):
    tmp2 = sum(B[j,i]*model.x[i]for i in N_index_list)
    return tmp2 <= d[j]

Bは1次元の配列ですが、B[j, i]と2つのインデックスが渡されています。

Comments

[kk_nikmann]

ご指摘ありがとうございます。修正したらうまく回りました。