jhconning
5/2/2017 - 2:01 AM

pandas: list comprehension to drop columns that match condition

pandas: list comprehension to drop columns that match condition

def drop_cols(df):
    '''drop columns that don't vary'''
    df.drop([ col for col in df.columns if (len(df[col].unique()) == 1)], 
        axis=1, inplace=True)