pythonで次のコードを簡単にしたい。

Question

次のような計算をしています。
Xというのは１次元配列で、
X = np.array([1,2,4,3,1,3,2,4,6,4,2,…])
のような感じです。

dim = 4
input_vec = np.zeros((len(X),dim+1))

Sum1 = np.zeros((dim+1,dim+1))
Sum2 = np.c_[np.zeros(dim+1)] #縦ベクトルにしておく
	
	for j in range(0,len(X)):
		input_vec[j] = np.array([[X[j]**0,X[j]**1,X[j]**2,X[j]**3,X[j]**4]])

		Sum1 = Sum1 + np.dot(input_vec[[j]].transpose(),input_vec[[j]])
		Sum2 = Sum2 + Y[j]*input_vec[[j]].transpose()

dim = 4 の場合はこのままでも良いのですが、dimを書き換えた時にちゃんと動くように、

input_vec = np.zeros((len(X),dim+1))

Sum1 = np.zeros((dim+1,dim+1))
Sum2 = np.c_[np.zeros(dim+1)] #縦ベクトルにしておく
	
	for j in range(0,len(X)):
		for i in range(0,dim+1):
			input_vec[j][i] = X[j]**i

		Sum1 = Sum1 + np.dot(input_vec[[j]].transpose(),input_vec[[j]])
		Sum2 = Sum2 + Y[j]*input_vec[[j]].transpose()

と書き換えましたが、エラーがでます。なぜでしょう。

エラー内容：
IndexError: index 1 is out of bounds for axis 0 with size 1

エラーがでる行は
input_vec[j][i] = X[j]**i
の行です。

Comments

[y_waiwai]

どの行でそのエラーが出るんでしょうか。質問は編集できますんで、編集で修正してください

[tiitoi]

まず記載コードの意図、つまり入力がなにで、出力としてどのような値を計算したいのかを説明していただけますか。

[physics303]

入力はX = np.array([1,2,4,3,1,3,2,4,6,4,2,…])です。出力はSum1とSum2です。

Answer 1

ふたつあるfor文の中で配列の大きさより大きいindexを指定しているのかと思います。for文の範囲を確認するのと、配列の大きさが0の場合(今回はおそらくない?)の例外処理をする必要があるかと思います。

Comments

[physics303]

ありがとうございます。
配列の大きさは len(X) × (dim + 1) で forループは (0 〜 len(x))×（0 〜(dim + 1)）で回しているので配列の大きさより大きいindexを指定しているってことはないと思うのですが…

[yskk]

(0 〜 len(x))が悪いようなきがします。xの要素数が10だとすると範囲は0~9になるはずです。 (0 〜 len(x)-1)ではないですか。dim+1も同様に、dim+1-1

[yskk]

ごめんなさい、他の言語のrangeの挙動と混ざってるかもです。違ってたらすみません

[yskk]

いずれにせよ input_vecの範囲外にアクセスしてるのは確かだと思うので、input_vecに期待通りの値が入っているか、エラーがでるのはi,jがそれぞれ何番のときかを確認してください