Pandas Practice Set-1: Set an existing column as the index of diamonds DataFrame and restore the index name, and move the index back to a column

Last update on May 03 2025 13:08:35 (UTC/GMT +8 hours)

44. Set Column as Index and Restore the Index Name

Write a Pandas program to set an existing column as the index of diamonds DataFrame and restore the index name, and move the index back to a column.

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("\nNew Dataframe:")
diamonds.set_index('color', inplace=True)
print(diamonds.head())
print("\nNow restore the index name:")
diamonds.index.name = 'color'
diamonds.reset_index(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

New Dataframe:
       carat      cut clarity  depth  table  price     x     y     z
color                                                               
E       0.23    Ideal     SI2   61.5   55.0    326  3.95  3.98  2.43
E       0.21  Premium     SI1   59.8   61.0    326  3.89  3.84  2.31
E       0.23     Good     VS1   56.9   65.0    327  4.05  4.07  2.31
I       0.29  Premium     VS2   62.4   58.0    334  4.20  4.23  2.63
J       0.31     Good     SI2   63.3   58.0    335  4.34  4.35  2.75

Now restore the index name:
  color  carat      cut clarity  depth  table  price     x     y     z
0     E   0.23    Ideal     SI2   61.5   55.0    326  3.95  3.98  2.43
1     E   0.21  Premium     SI1   59.8   61.0    326  3.89  3.84  2.31
2     E   0.23     Good     VS1   56.9   65.0    327  4.05  4.07  2.31
3     I   0.29  Premium     VS2   62.4   58.0    334  4.20  4.23  2.63
4     J   0.31     Good     SI2   63.3   58.0    335  4.34  4.35  2.75

For more Practice: Solve these Related Problems:

Write a Pandas program to set a column as the index of the diamonds DataFrame, then restore the original index name after resetting.
Write a Pandas program to assign a column as index, rename the index, and later convert the index back into a column.
Write a Pandas program to set an existing column as index, then reset the index and ensure the index name is retained as a new column.
Write a Pandas program to switch a column to the index, perform a sort, and then move the index back into the columns with its name restored.

Go to:

Previous:Write a Pandas program to set an existing column as the index of diamonds DataFrame.
Next: Write a Pandas program to access a specified Series index and the Series values of diamonds DataFrame.

Have another way to solve this solution? Contribute your code (and comments) through Disqus.