Pandas DataFrame: Delete the 'attempts' column from the DataFrame

Last update on December 21 2024 07:41:55 (UTC/GMT +8 hours)

Write a Pandas program to delete the 'attempts' column from the DataFrame.

Sample DataFrame:
exam_data = {'name': ['Anastasia', 'Dima', 'Katherine', 'James', 'Emily', 'Michael', 'Matthew', 'Laura', 'Kevin', 'Jonas'],
'score': [12.5, 9, 16.5, np.nan, 9, 20, 14.5, np.nan, 8, 19],
'attempts': [1, 3, 2, 3, 2, 3, 1, 1, 2, 1],
'qualify': ['yes', 'no', 'yes', 'no', 'no', 'yes', 'yes', 'no', 'no', 'yes']}
labels = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j']

Sample Solution :

Python Code :

import pandas as pd
import numpy as np
exam_data  = {'name': ['Anastasia', 'Dima', 'Katherine', 'James', 'Emily', 'Michael', 'Matthew', 'Laura', 'Kevin', 'Jonas'],
        'score': [12.5, 9, 16.5, np.nan, 9, 20, 14.5, np.nan, 8, 19],
        'attempts': [1, 3, 2, 3, 2, 3, 1, 1, 2, 1],
        'qualify': ['yes', 'no', 'yes', 'no', 'no', 'yes', 'yes', 'no', 'no', 'yes']}
labels = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j']
df = pd.DataFrame(exam_data , index=labels)
print("Original rows:")
print(df)
print("\nDelete the 'attempts' column from the data frame:")
df.pop('attempts')
print(df)

Sample Output:

Original rows:                                                          
   attempts       name qualify  score                                  
a         1  Anastasia     yes   12.5                                  
b         3       Dima      no    9.0                                  
c         2  Katherine     yes   16.5                                  
d         3      James      no    NaN                                  
e         2      Emily      no    9.0                                  
f         3    Michael     yes   20.0                                  
g         1    Matthew     yes   14.5                                  
h         1      Laura      no    NaN                                  
i         2      Kevin      no    8.0                                  
j         1      Jonas     yes   19.0                                  
                                                                       
Delete the 'attempts' column from the data frame:                      
        name qualify  score                                            
a  Anastasia     yes   12.5                                            
b       Dima      no    9.0                                            
c  Katherine     yes   16.5                                            
d      James      no    NaN 
e      Emily      no    9.0                                            
f    Michael     yes   20.0                                            
g    Matthew     yes   14.5                                            
h      Laura      no    NaN                                            
i      Kevin      no    8.0                                            
j      Jonas     yes   19.0                    

Explanation:

The above code first creates a Pandas DataFrame named ‘df’ using the provided ‘exam_data’ dictionary and ‘labels’ list as the index.

df.pop('attempts'): This line removes the 'attempts' column from the DataFrame using the pop() method, which returns the removed column as a Pandas Series.

Finally print() function prints the updated 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 change the name 'James' to 'Suresh' in name column of the data frame.
Next: Write a Pandas program to insert a new column in existing DataFrame.