[Pandas] Check if a column exists in a DataFrame

python Share on :

We will learn in this post to check if a column exists in a Pandas DataFrame or not. We will write the condition to return true if the column exists and return false if the column does not exist.

import pandas as pd

employees = [
  ['Robin', 30, 'SOS', 'India'],
  ['Rick', 28, 'PSA', 'US'],
  ['Tony Stark', 32, 'COS', 'US']
]

# create dataframe
df = pd.DataFrame(employees, columns=['Name', 'Age', 'Dept', 'Country'])

# check if column exist in the dataframe
if 'Age' in df.columns:
  print("Coumn exists in the DataFrame")
else:
  print("Column does not exist in the DataFrame")

If you want to check whether a column exists in a Pandas DataFrame or not then you can use the methods explained in this post.

We are using the "in" keyword to check that. Below is the example to understand it..