Articles → PANDAS → Add and delete columns from DataFrame in pandas
Add and delete columns from DataFrame in pandas
Adding a new 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)
Click to Enlarge
Deleting a 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("\nDataFrame before deleting new column\n")
print(df)
df = df.drop('X', axis=1)
print("\nDataFrame after deleting new column\n")
print(df)
Click to Enlarge