Pandas: Check whether a given column is present in a DataFrame or not

Last update on April 03 2025 13:11:19 (UTC/GMT +8 hours)

46. Check Column Presence

Write a Pandas program to check whether a given column is present in a DataFrame or not.

Sample Solution :

Python Code :

import pandas as pd
d = {'col1': [1, 2, 3, 4, 7], 'col2': [4, 5, 6, 9, 5], 'col3': [7, 8, 12, 1, 11]}
df = pd.DataFrame(data=d)
print("Original DataFrame")
print(df)
if 'col4' in df.columns:
  print("Col4 is present in DataFrame.")
else:
  print("Col4 is not present in DataFrame.")
if 'col1' in df.columns:
  print("Col1 is present in DataFrame.")
else:
  print("Col1 is not present in DataFrame.")

Sample Output:

Original DataFrame
   col1  col2  col3
0     1     4     7
1     2     5     8
2     3     6    12
3     4     9     1
4     7     5    11
Col4 is not present in DataFrame.
Col1 is present in DataFrame.

Explanation:

The above code first creates a Pandas DataFrame ‘df’ with three columns named col1, col2, and col3 and five rows of data.

The code then checks if the column 'col4' is present in the DataFrame using the in operator with the columns attribute of the DataFrame. Since 'col4' is not one of the DataFrame columns, the output will be "Col4 is not present in DataFrame."

Next, the code checks if the column 'col1' is present in the DataFrame. Since 'col1' is one of the DataFrame columns, the output will be "Col1 is present in DataFrame."

For more Practice: Solve these Related Problems:

Write a Pandas program to check if a specified column exists in a DataFrame and then output a boolean result.
Write a Pandas program to verify the presence of multiple columns in a DataFrame and then list the missing columns.
Write a Pandas program to test if a column exists in a DataFrame and then conditionally perform an operation based on that check.
Write a Pandas program to determine the presence of a column and then output a custom message if it is missing.

Python-Pandas Code Editor:

Have another way to solve this solution? Contribute your code (and comments) through Disqus.

Previous: Write a Pandas program to find the row for where the value of a given column is maximal.
Next: Write a Pandas program to get the specified row value of a given DataFrame.