df.pop()
可以使用这个来删除某一列(不能是多列),只有一个参数,就是列名,可以是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 '''