热线电话:13121318867

登录
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']]
11.0829
4
关注作者
收藏
评论(0)

发表评论

暂无数据
推荐帖子