Python使用drop()方法?

时间:2017-05-04 21:12:20

标签: python pandas dataframe

我们说我有一个包含以下列的DataFrame:TAGALIASCOMMENTTYPE

有没有办法放弃所有不是COMMENT的列而不必键入以下行?

df.drop(["TAG","ALIAS","TYPE"], 1)

有没有办法在那里的某处键入if语句并删除任何不是列调用的内容,例如COMMENT

2 个答案:

答案 0 :(得分:3)

如果你知道你只想要列COMMENT,那只是为了

df = df['COMMENT']

如果您正在寻找以COMMENT开头的各种列,比如COMMENT1,COMMENT2等,您可以使用过滤器

df = df.filter(like = 'COMMENT')

正如@piRsquared建议的那样,选择列的另一种方法是

df = df[['COMMENT']]

如果您想选择多个列

,则尤其需要这样做
df = df[['COMMENT', 'COMMENT1']]

答案 1 :(得分:3)

我在添加答案时改变了主意。这是令人讨厌的,因为A-Za-z和ayhan所说的更有意义......

......但是,这并不能阻止我发布这个

df.drop(df.columns.difference(['COMMENT']), 1)