gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-12-01 08:13:23 25 4
gpt4 key购买 nike

我有一个作为 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']]

关于python - 将数据拆分为特征和标签后,标签列形状不一致,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55096091/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com