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