时间戳

首页 > 解决方案 > 如何将数据框拆分为 x 和 y

python - 如何将数据框拆分为 x 和 y

问题描述

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

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]

然而，经过许多变化和谷歌搜索，我无法解决这条线给我的这个错误，也无法理解它：

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

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

标签： pythonnumpy

解决方案

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

推荐阅读