您的位置: 首页 > IT文章 > df.pop() 删除某一列（不能是多列） df.pop()

df.pop() 删除某一列（不能是多列） df.pop()

分类: IT文章 • 2023-11-12 13:18:00

可以使用这个来删除某一列（不能是多列），只有一个参数，就是列名，可以是str类型，函数返回的是被删除的列，df直接是删除后的df，不需要我们处理

我们建模时，需要单独保留某个特征，比如y值，我们就可以用y=xx.pop(label)了，很实用。

例子

df = pd.DataFrame([('falcon', 'bird', 389.0),
                   ('parrot', 'bird', 24.0),
                   ('lion', 'mammal', 80.5),
                   ('monkey', 'mammal', np.nan)],
                  columns=('name', 'class', 'max_speed'))
df
'''
     name   class  max_speed
0  falcon    bird      389.0
1  parrot    bird       24.0
2    lion  mammal       80.5
3  monkey  mammal        NaN
'''

删除某列

df.pop('class')
'''
0      bird
1      bird
2    mammal
3    mammal
Name: class, dtype: object
'''

删除后的df

df
'''
     name  max_speed
0  falcon      389.0
1  parrot       24.0
2    lion       80.5
3  monkey        NaN
'''