无论是否存在所有列名,都从pandas数据帧中删除列

时间:2014-04-22 14:28:37

标签: python filter pandas dataframe

我想使用df.drop(drop_list, axis=1)从我的数据框中删除多个列,但是如果drop_list中没有df.columns.tolist()的条目,则命令失败,怎么能我阻止这种情况发生?

2 个答案:

答案 0 :(得分:3)

df.drop(set(drop_list) & set(df.columns), axis=1)

答案 1 :(得分:1)

我认为这应该有效:

df.drop(set.intersection(set(drop_list), df.columns.tolist()), axis=1)

(EdChum评论后编辑)