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

Question

我有一个数据集作为pandas数据框，需要分为功能集和标签。截至目前，我将各列划分如下，

features = df2.drop('case_of_injury_group', 1)
labels = df2['case_of_injury_group']

但是标签的形状不符合我的预期

features.shape

给出（39778，12）和

labels.shape

给出（39778，），但我希望它是（39778，1）。请让我知道我在做什么错

Answer 1

如果要使用一列DataFrame，请按一个元素嵌套列表进行选择：

labels = df2[['case_of_injury_group']]