将数据拆分为要素和标签后，标签列形状不一致_CDA答疑社区

热线电话：13121318867

登录

啊啊啊啊啊吖

2019-03-11 阅读量: 1067

将数据拆分为要素和标签后，标签列形状不一致

我有一个数据集作为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

1

4

关注作者

收藏

评论(0)

发表评论

暂无数据

CDA考试动态

CDA报考指南

推荐帖子