我们说我有一个包含以下列的DataFrame:TAG
,ALIAS
,COMMENT
,TYPE
。
有没有办法放弃所有不是COMMENT
的列而不必键入以下行?
df.drop(["TAG","ALIAS","TYPE"], 1)
有没有办法在那里的某处键入if语句并删除任何不是列调用的内容,例如COMMENT
?
答案 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)