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

Comments

[y_waiwai]

どこでどういうエラーが出るんでしょうか。エラーメッセージを提示してください

Answer 1

for j in ... がインデントされているのが問題ではないかと。