如何将数据帧分为x和y

时间:2018-12-31 20:18:43

标签: python numpy

我将数据分为训练数据和测试数据,如下所示:

train, test = train_test_split(dataFrame(), test_size=0.2)

哪些奇迹,我的训练数据框架如下:

     PassengerId  Survived  SibSp  Parch
77            78         0      0      0
748          749         0      1      0
444          445         1      0      0
361          362         0      1      0
576          577         1      0      0
27            28         0      3      2
232          233         0      0      0
424          425         0      1      1
785          786         0      0      0
…            …           …      …      … 

我现在尝试获取X和Y列,X是我的SibSp列,Y是我的Parch列。在遵循许多tutorials on Regression并训练了我的AI之后,每个人都将列“拆分”为x和y,如下所示:

x = train[:, 0:2]

但是,经过许多变化和谷歌搜索后,我无法解决此错误,这行给我的也无法理解:

  

TypeError:不可散列的类型:'sl​​ice'

如何在训练数据框中将SibSp列拆分为x的数组并将Parch列拆分为y的数组?

1 个答案:

答案 0 :(得分:3)

切片的正确方法是x = train.iloc[:, 0:2]