如何解决“ ValueError：找到输入样本数量不一致的输入变量：[395，1]”的Python错误

Question

我是python的新手，我正在尝试学习机器学习。然后我遇到这种错误。我该怎么办？

predict = "G3"

X = np.array(data.drop([predict], 1))
y = np.array([predict]) 
x_train, x_test, y_train, y_test = sklearn.model_selection.train_test_split(X, y, test_size = 0.1)