w3resource

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


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

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.

What is the difficulty level of this exercise?

Test your Programming skills with w3resource's quiz.



Follow us on Facebook and Twitter for latest update.