w3resource logo

:

Python Exercises

Python Pandas: Insert a new column in existing DataFrame

Python Pandas: DateFrame Exercise-20 with Solution

Write a Python program to insert a new column in existing DataFrame.

Sample DateFrame:
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']
Insert the following column:
color = ['Red','Blue','Orange','Red','White','White','Blue','Green','Green','Red']

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("Orginal rows:")
print(df)
color = ['Red','Blue','Orange','Red','White','White','Blue','Green','Green','Red']
df['color'] = color
print("\nNew DataFrame after inserting the 'color' column")
print(df)

Sample Output:

Orginal 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                                  
                                                                       
New DataFrame after inserting the 'color' column                       
   attempts       name qualify  score   color                          
a         1  Anastasia     yes   12.5     Red                          
b         3       Dima      no    9.0    Blue                          
c         2  Katherine     yes   16.5  Orange                          
d         3      James      no    NaN     Red
e         2      Emily      no    9.0   White                          
f         3    Michael     yes   20.0   White                          
g         1    Matthew     yes   14.5    Blue                          
h         1      Laura      no    NaN   Green                          
i         2      Kevin      no    8.0   Green                          
j         1      Jonas     yes   19.0     Red                    

Python Code Editor:

Improve this sample solution and post your code through Disqus