Gyan Sangrah

Add and delete columns from DataFrame in pandas

In this article, we will add and delete columns from DataFrame in pandas

Adding a new column

Consider the following code for adding the column

import numpy as np
import pandas as pd


df = pd.DataFrame(np.random.randn(5,4),['A', 'B', 'C', 'D', 'E'],['W', 'X', 'Y', 'Z'])

print("DataFrame before adding a new column\n")
print(df)

# Adding a new column
df['New Column'] = df['W'] + df['X']

print("\nDataFrame after adding a new column\n")
print(df)

Here is the output

In the above example, I have created a ‘New Column’ by adding the values in column ‘X’ and ‘Y’.

Deleting a column

To delete a column, you can use the drop method. Consider the following example

import numpy as np
import pandas as pd


df = pd.DataFrame(np.random.randn(5,4),['A', 'B', 'C', 'D', 'E'],['W', 'X', 'Y', 'Z'])



print("\nDataFrame before deleting new column\n")
print(df)

df = df.drop('X', axis=1)

print("\nDataFrame after deleting new column\n")
print(df)

Here axis = 0 means index and axis = 1 is columns. Here is the output

Posted By -	Karan Gupta

Posted On -	Friday, April 12, 2019

Query/Feedback

Your Email Id

Subject

Query/Feedback	Characters remaining 250