ValueError when trying to separate training and test set

Question

I\'ve tried the following code in python:

X=df[["age","bmi","children"]].values.reshape(-1,1)
Y=df["charges"].values.r