Pandas Dataframe- Drop columns if all the values with the column are either 0,1,nan

Question

Assuming this is the dataframe. I looking to drop the columns if all the values within columns are either 0,1, NaN.

df = pd.DataFrame([[1,0,0,0], [0,0,1,0],[2,'NaN',1,0]])

End result should be just first column "0" and drop remaining columns.

Answer 1

Try:

lst = [0, 1, 'NaN']

mask = df.isin(lst).all(axis=0)
df.drop(mask.loc[mask].index, axis=1, inplace=True)

it will essentially ensure you drop all columns, for which all values are in lst .

Answer 2

use this:

df.drop([1,2])

See also the documentation