1

2

3

4

5

6

7

8

9

10

11

# 从X和Y中取出相应步长对应的数组并保存至x_data和y_data中

x_data = []

y_data = []

for i in range(len(K) - 24*n + 1):

    x_data.append(X[i:i + 24*n, :])

    y_data.append(Y[i + 24 * n:i + 24 * n + 24 * n, :])

# 将x_data和y_data转为ndarray

n_train_hours = 24*365

x_train = np.array(x_data)[: n_train_hours, :, :]

y_train = np.array(y_data)[: n_train_hours, :, :]

12345678910

1

2

3

4

# 检查y_data中的每个二维数组的shape

for i in range(len(y_data)):

    print(y_data[i].shape)

123

1

2

3

4

5

6

x_data = []

y_data = []

for i in range(len(Y) - 24 * n - 24 * n + 1):

    x_data.append(X[i:i + 24*n, :])

    y_data.append(Y[i + 24*n:i + 24*n + 24*n, :])

12345