arrays for test and train set for CNN using python

I have a set of 4 images that i convert them to arrays to train a CNN

i found this method that says:

X_train, X_test, y_train, y_test = train_test_split(
...     X, y, test_size=0.33, random_state=42)
...

knowing that X_train is usually a full image (array) and y_train is a label that would tells us how the image is classified

i extracted 4 label arrays to satisfy the x train and train conditions

Now I 'm stuck in how can I put them. I mean x train is just one value and I have 4 arrays same for y train ( for the testalso I have 2 arrays for x test and y test)

Can anyone propose a solution?