Python 蟒蛇熊猫：I'；我在使用drop函数的行中出现了一个错误

我开始学习机器学习，我跟随youtube上的某个人学习。 以下是链接： 所以基本上，这和他写的代码是一样的

import pandas as pd
import numpy as np
import sklearn
from sklearn import linear_model
from sklearn.utils import shuffle

data = pd.read_csv("student-mat.csv", sep=";")

#print(data.head())
data = data[["G1", "G2", "G3", "studytime", "failures", "absences"]]
print(data.head())
predict = "G3"

x = np.array(data.drop([predict]), axis=1)
y = np.array(data[predict])

x_train, x_test, y_train, y_test = sklearn.model_selection.train_test_split(x, y, test_size=0.1)
linear = linear_model.LinearRegression()

linear.fit(x_train, y_train)
acc = linear.score(x_test, y_test)
print(acc)

当我运行这个

Traceback (most recent call last):
  File "C:/Users/asb46/PycharmProjects/tensorEnv/lec1.py", line 14, in <module>
    x = np.array(data.drop([predict]), axis=1)
  File "C:\Users\asb46\Anaconda3\envs\tensor\lib\site-packages\pandas\core\frame.py", line 3697, in drop
    errors=errors)
  File "C:\Users\asb46\Anaconda3\envs\tensor\lib\site-packages\pandas\core\generic.py", line 3111, in drop
    obj = obj._drop_axis(labels, axis, level=level, errors=errors)
  File "C:\Users\asb46\Anaconda3\envs\tensor\lib\site-packages\pandas\core\generic.py", line 3143, in _drop_axis
    new_axis = axis.drop(labels, errors=errors)
  File "C:\Users\asb46\Anaconda3\envs\tensor\lib\site-packages\pandas\core\indexes\base.py", line 4404, in drop
    '{} not found in axis'.format(labels[mask]))
KeyError: "['G3'] not found in axis"

回溯（最近一次呼叫最后一次）：
文件“C:/Users/asb46/PycharmProjects/tensorEnv/lec1.py”，第14行，在
x=np.array（data.drop（[predict]），axis=1）
文件“C:\Users\asb46\Anaconda3\envs\tensor\lib\site packages\pandas\core\frame.py”，第3697行，在下拉列表中
错误=错误）
文件“C:\Users\asb46\Anaconda3\envs\tensor\lib\site packages\pandas\core\generic.py”，第3111行，在下拉列表中
obj=obj.\u下降\u轴（标签、轴、级别=级别、错误=错误）
文件“C:\Users\asb46\Anaconda3\envs\tensor\lib\site packages\pandas\core\generic.py”，第3143行，在放置轴中
新建_轴=轴。放置（标签，错误=错误）
文件“C:\Users\asb46\Anaconda3\envs\tensor\lib\site packages\pandas\core\indexes\base.py”，第4404行，在下拉列表中
在axis.格式（标签[mask]）中找不到“{}”
KeyError:“['G3']未在轴中找到”

我正在使用pandas:0.23.4版本。我尝试降级，但仍然出现错误。。。我不知道

---

我的代码的正确版本是什么

对代码稍加修改即可解决此问题

而不是

x = np.array(data.drop([predict]), axis=1)

做

说明：axis是drop方法的参数，而不是np.array的参数（data.drop（[predict]，axis=1））？

x = np.array(data.drop([predict], axis=1))