Python Pandas - Rearrange levels in MultiIndex

PythonServer Side ProgrammingProgramming

To rearrange levels in MultiIndex, use the MultiIndex.reorder_levels() method in Pandas. Set the order of levels using the order parameter.

At first, import the required libraries −

import pandas as pd

MultiIndex is a multi-level, or hierarchical, index object for pandas objects. Create arrays −

arrays = [[2, 4, 3, 1], ['Peter', 'Chris', 'Andy', 'Jacob'], [50, 30, 40, 70]]

The "names" parameter sets the names for each of the index levels. The from_arrays() is used to create a MultiIndex −

multiIndex = pd.MultiIndex.from_arrays(arrays, names=('rank', 'student', 'points'))

Reorder levels of MultiIndex. The "order" parameter is used to set the level in a form to reorder levels

print("\nReorder levels in MultiIndex...\n",multiIndex.reorder_levels(order=[2, 0, 1]))

Example

Following is the code −

import pandas as pd

# MultiIndex is a multi-level, or hierarchical, index object for pandas objects
# Create arrays
arrays = [[2, 4, 3, 1], ['Peter', 'Chris', 'Andy', 'Jacob'], [50, 30, 40, 70]]

# The "names" parameter sets the names for each of the index levels
# The from_arrays() is used to create a MultiIndex
multiIndex = pd.MultiIndex.from_arrays(arrays, names=('rank', 'student', 'points'))

# display the MultiIndex
print("The MultiIndex...\n",multiIndex)

# get the levels in MultiIndex
print("\nThe levels in MultiIndex...\n",multiIndex.levels)

# Reorder levels of MultiIndex
# The "order" parameter is used to set the level in a form to reorder levels
print("\nReorder levels in MultiIndex...\n",multiIndex.reorder_levels(order=[2, 0, 1]))

Output

This will produce the following output −

The MultiIndex...
MultiIndex([(2, 'Peter', 50),
            (4, 'Chris', 30),
            (3,  'Andy', 40),
            (1, 'Jacob', 70)],
            names=['rank', 'student', 'points'])

The levels in MultiIndex...
   [[1, 2, 3, 4], ['Andy', 'Chris', 'Jacob', 'Peter'], [30, 40, 50, 70]]

Reorder levels in MultiIndex...
MultiIndex([(50, 2, 'Peter'),
            (30, 4, 'Chris'),
            (40, 3, 'Andy'),
            (70, 1, 'Jacob')],
            names=['points', 'rank', 'student'])
raja
Published on 19-Oct-2021 08:08:47
Advertisements