How to check if a column exists in Pandas?

To check if a column exists in a Pandas DataFrame, we can take the following Steps −

Steps

• Create a two-dimensional, size-mutable, potentially heterogeneous tabular data, df.

• Print the input DataFrame, df.

• Initialize a col variable with column name.

• Create a user-defined function check() to check if a column exists in the DataFrame.

• Call check() method with valid column name.

• Call check() method with invalid column name.

Example

 Live Demo

import pandas as pd

def check(col):
   if col in df:
      print "Column", col, "exists in the DataFrame."
   else:
      print "Column", col, "does not exist in the DataFrame."

df = pd.DataFrame(
   {
      "x": [5, 2, 1, 9],
      "y": [4, 1, 5, 10],
      "z": [4, 1, 5, 0]
   }
)

print "Input DataFrame is:\n", df
col = "x"
check(col)
col = "a"
check(col)

Output

Input DataFrame is:
   x  y  z
0  5  4  4
1  2  1  1
2  1  5  5
3  9 10  0
Column x exists in the DataFrame.
Column a does not exist in the DataFrame.

Rishikesh Kumar Rishi

Published on 30-Aug-2021 12:10:22

• Related Questions & Answers
• How to check if a MySQL database exists?
• Check if a value exists in a column in a MySQL table?
• How to check if a variable exists in JavaScript?
• How to check if a file exists in Golang?
• How do I check to see if a column name exists in a CachedRowSet in JDBC?
• Check if a file exists in Java
• Check if a File exists in C#
• How to check if a key exists in a Python dictionary?
• Check if MongoDB database exists?
• How to check if a vector exists in a list in R?
• How to check if a key exists in a map in Golang?
• How to check if a file exists or not in Java?
• How to check if an item exists in a C# array?
• How do I check if a Python variable exists?
• How to use Lua Programming to check if a file exists?