Check whether a given column is present in a Dataframe

DataFrame is a structure that contains 2-dimensional data and its corresponding labels. DataFrame.columns attribute return the column labels of the given Dataframe . In Order to check if a column exists in Pandas DataFrame, you can use "in" expression.

 
import pandas as pd
import numpy as np
df = pd.DataFrame()
df['Name'] = ['John', 'Doe', 'Bill']
df['Promoted'] = [True, False,True]
df['Marks'] = [82, 38, 63]
df

 
   Name  Promoted  Marks
0  John      True     82
1   Doe     False     38
2  Bill      True     63

Use in

 
if 'Promoted' in df.columns:
    print("Yes")
else:
    print("No")

Yes

Use not in

 
if 'Promoted' not in df.columns:
    print("Yes")
else:
    print("No")

No

Check for multiple columns exists in DataFrame

 
if set(['Name','Promoted']).issubset(df.columns):
    print("Yes")
else:
    print("No")

Yes

How to Check whether a given column is present in a Dataframe

You can alternatively be constructed with curly braces {}.

 
if not {'Name', 'Promoted'}.issubset(df.columns):
    print("Yes")
else:
    print("No")

No

Next : How to delete column from pandas DataFrame