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

我有一个数据集作为数据框架，需要将其划分为要素集和标签。到目前为止，我正在按如下方式划分各列：

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

但是标签的形状不是我想象的那样

features.shape

给出（39778,12）和

---

给出（39778，）但我希望它是（39778，1）。如果需要一列

DataFrame

select by one element嵌套列表，请让我知道我在这里做错了什么：

labels = df2[['case_of_injury_group']]

---

工作。谢谢

labels = df2[['case_of_injury_group']]