Python Pandas - Set levels in a MultiIndex

PythonServer Side ProgrammingProgramming

To set levels in a MultiIndex, use the MultiIndex.set_levels() method in Pandas. 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 = [[1, 2, 3, 4], ['John', 'Tim', 'Jacob', 'Chris']]

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=('ranks', 'student'))

Set the levels in MultiIndex −

print("\nSet new levels in Multi-index...\n",multiIndex.set_levels([['p', 'q', 'r', 's'], [10, 20, 30, 40]]))

Example

Following is the code −

import pandas as pd

# MultiIndex is a multi-level, or hierarchical, index object for pandas objects
# Create arrays
arrays = [[1, 2, 3, 4], ['John', 'Tim', 'Jacob', 'Chris']]

# 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=('ranks', 'student'))

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

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

# set the levels in MultiIndex
print("\nSet new levels in Multi-index...\n",multiIndex.set_levels([['p', 'q', 'r', 's'], [10, 20, 30, 40]]))

Output

This will produce the following output −

The Multi-index...
MultiIndex([(1,  'John'),
            (2,   'Tim'),
            (3, 'Jacob'),
            (4, 'Chris')],
            names=['ranks', 'student'])

The levels in Multi-index...
   [[1, 2, 3, 4], ['Chris', 'Jacob', 'John', 'Tim']]

Set new levels in Multi-index...
MultiIndex([('p', 30),
            ('q', 40),
            ('r', 20),
            ('s', 10)],
            names=['ranks', 'student'])
raja
Updated on 19-Oct-2021 07:11:01

Advertisements