How can I integrade splitted test/train data and splitted test/train labels on this code?

Question

I have code that its manually splitting data but actually in my case I have splitter data already:

train_features = (1561,2) - dtype float64,

test_features = (