2019-03-11
阅读量:
937
将数据拆分为要素和标签后,标签列形状不一致
我有一个数据集作为pandas数据框,需要分为功能集和标签。截至目前,我正在将列分割如下,
features = df2.drop('case_of_injury_group', 1)
labels = df2['case_of_injury_group']
但标签的形状并不像我预期的那样
features.shape
给(39778,12)和
labels.shape
给(39778,)但我想要它(39778,1)。请告诉我这里我做错了什么
解决办法:如果想要一列DataFrame
选择一个元素嵌套列表:
labels = df2[['case_of_injury_group']]






评论(0)


暂无数据
推荐帖子
0条评论
0条评论
0条评论