Та же операция обеспечивает две различные формы в течение l oop python

Question

Я перебираю 8 разных списков, выполняя одну и ту же операцию в каждом списке, но в конце я получаю одну фигуру в первой, а затем другую фигуру в 7 следующих. Вот код:

def clean_data(X, Y, flash):
  cleaned_Xtrain = []
  cleaned_Xtest  = []
  cleaned_Ytrain = []
  cleaned_Ytest  = []
  for Xi, flash_i in zip(X, flash):

    X_samples = [np.array(Xi[i[0]:i[0]+351]) for i in flash_i]
    column    = [i[2] for i in flash_i]
    label     = [i[3] - 1 for i in flash_i]

    X_train = np.array(X_samples.copy())
    X_test  = np.array(X_samples.copy())
    X_train = np.array(X_train[:1800])
    X_test  = np.array(X_test)
    X_test  = np.asarray(X_test[1800:])

    print(X_train.shape, X_test.shape, X_test[0].shape)

    y_train = np.array(to_categorical(label[:1800]))
    y_test  = np.array(to_categorical(label[1800:]))
    col_train = np.array(column[:1800])
    col_test  = np.array(column[1800:])

    cleaned_Xtrain.append(X_train)
    cleaned_Xtest .append(X_test)

  return cleaned_Xtrain, cleaned_Xtest, y_train, y_test, col_train, col_test

И формы (пожалуйста, игнорируйте 4 первых пустых списка):

(1800, 351) (2400, 351) (351,)
(1800,) (2400,) (351,)
(1800,) (2400,) (351,)
(1800,) (2400,) (351,)
(1800,) (2400,) (351,)
(1800,) (2400,) (351,)
(1800,) (2400,) (351,)
(1800,) (2400,) (351,)