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")

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")

Next : How to delete column from pandas DataFrame

Related Topics