IndexError：創建訓練測試拆分時數組的索引過多

Question

我正在創建一個神經網絡，目前正在研究； train, test split但出現錯誤IndexError: too many indices for array我的代碼是：

import csv
import math
import numpy as np 
import pandas as pd
from keras.models import Sequential
from keras.layers import Dense
import datetime
import matplotlib.pyplot as plt
from sklearn.model_selection import train_test_split

X1 = Values[1:16801] #16,800 values
Y1 = P1[1:16801]#16,800 values

train_size = int(len(X1) * 0.67)
test_size = len(X1) - train_size

train, test = X1[0:train_size,], X1[train_size:len(X1),]
def Data(X1, look_back=1):
     dataX, dataY = [], []
     for i in range(len(X1)-look_back-1):
         a = X1[i:(i+look_back), 0]
         dataX.append(a)
         dataY.append(Y1[i + look_back, 0])
     return numpy.array(dataX), numpy.array(dataY)

look_back = 1
trainX, testX = Data(train, look_back)

testX, testY = Data(test, look_back)

look_back = 1
trainX, testX = Data(train, look_back)

testX, testY = Data(test, look_back)

我有X1的16,800個值，如下所示：

[0.03454225 0.02062136 0.00186715 ... 0.92857565 0.64930691 0.20325924]

我的Y1數據看起來像： [ 2.25226244 1.44078451 0.99174488 ... 12.8397099 9.75722427 7.98525797]

我的回溯錯誤消息是：

IndexError                                Traceback (most recent call last)
<ipython-input-11-afedcaa56e0b> in <module>()
     86 
     87 look_back = 1
---> 88 trainX, testX = Data_split(train, look_back)
     89 
     90 testX, testY = Data_split(test, look_back)

<ipython-input-11-afedcaa56e0b> in Data(X1, look_back)
     78     dataX, dataY = [], []
     79     for i in range(len(X1)-look_back-1):
---> 80         a = X1[i:(i+look_back), 0]
     81         dataX.append(a)
     82         dataY.append(Y1[i + look_back, 0])

IndexError: too many indices for array

我之前曾問過一個非常類似的問題並獲得了答案，但不幸的是，我無法對該錯誤應用該解決方案

Answer 1

問題出在數組的維度上。 您正在嘗試使用不存在的多個維度索引訪問元素。 看第80行。

 a = X1[i:(i+look_back), 0]   in your case metrics is just single dimention.

樣本2D指標表示形式（，）

“，”是對具有行和列的二維數組的引用，但是不幸的是，您將X1作為ndarray。

[0.03454225 0.02062136 0.00186715 ... 0.92857565 0.64930691 0.20325924]

類似的問題示例：-

>>> np.ndarray(4)
array([2.0e-323, 1.5e-323, 2.0e-323, 1.5e-323])
>>> a[1:2,0]
Traceback (most recent call last):
  File "<pyshell#38>", line 1, in <module>
    a[1:2,0]
IndexError: too many indices for array
>>> a[1:2]
array([-2.68156159e+154])
>>>