如何将数据帧拆分为 x 和 y

Question

我将数据拆分为训练数据和测试数据，如下所示：

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 列。在学习了许多关于回归的教程并训练了我的 AI 之后，每个人都像这样将列“拆分”为 x 和 y：

x = train[:, 0:2]

但是，经过多次变化和谷歌搜索后，我无法解决此行给我的错误，也无法理解它：

类型错误：不可散列的类型：“切片”

如何在训练数据框中将 SibSp 列拆分为 x 数组，将 Parch 列拆分为 y 数组？

Answer 1

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