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 '''