Pandas Practice Set-1: Remove multiple columns at once of the diamonds Dataframe
10. Remove Multiple Columns at Once
Write a Pandas program to remove multiple columns at once of the diamonds Dataframe.
Sample Solution :
Python Code :
import pandas as pd
diamonds = pd.read_csv('https://raw.githubusercontent.com/mwaskom/seaborn-data/master/diamonds.csv')
print("Original Dataframe:")
print(diamonds.head())
print("\nRemove multiple columns:")
diamonds.drop(['carat', 'color', 'y'], axis=1, inplace=True)
print(diamonds.head())
Sample Output:
Original Dataframe:
carat cut color clarity depth table price x y z
0 0.23 Ideal E SI2 61.5 55.0 326 3.95 3.98 2.43
1 0.21 Premium E SI1 59.8 61.0 326 3.89 3.84 2.31
2 0.23 Good E VS1 56.9 65.0 327 4.05 4.07 2.31
3 0.29 Premium I VS2 62.4 58.0 334 4.20 4.23 2.63
4 0.31 Good J SI2 63.3 58.0 335 4.34 4.35 2.75
Remove multiple columns:
cut clarity depth table price x z
0 Ideal SI2 61.5 55.0 326 3.95 2.43
1 Premium SI1 59.8 61.0 326 3.89 2.31
2 Good VS1 56.9 65.0 327 4.05 2.31
3 Premium VS2 62.4 58.0 334 4.20 2.63
4 Good SI2 63.3 58.0 335 4.34 2.75
For more Practice: Solve these Related Problems:
- Write a Pandas program to drop a list of specified columns from the diamonds DataFrame in a single operation.
- Write a Pandas program to remove multiple columns using the axis parameter and verify by checking the DataFrame shape.
- Write a Pandas program to drop non-numeric columns from the diamonds DataFrame and display the resulting DataFrame.
- Write a Pandas program to remove several columns using drop() with inplace=True and then print the updated column names.
Go to:
Previous: Write a Pandas program to remove the second column of the diamonds Dataframe.
Next: Write a Pandas program to remove multiple rows at once (axis=0 refers to rows) from diamonds dataframe.
Python Code Editor:
Have another way to solve this solution? Contribute your code (and comments) through Disqus.
What is the difficulty level of this exercise?