Python Pandas - Create a DataFrame with the levels of the MultiIndex as columns

PythonServer Side ProgrammingProgramming

To create a DataFrame with the levels of the MultiIndex as columns, use the to_frame() 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'))

Create a DataFrame with the levels of the MultiIndex as columns using to_frame() −

dataFrame = multiIndex.to_frame()

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)

# Create a DataFrame with the levels of the MultiIndex as columns using to_frame()
dataFrame = multiIndex.to_frame()

# Display the DataFrame
print("\nThe DataFrame...\n",dataFrame)

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']]

The DataFrame...
                ranks  student
ranks  student
1      John        1   John
2      Tim         2   Tim
3      Jacob       3   Jacob
4      Chris       4   Chris
raja
Published on 19-Oct-2021 07:30:34
Advertisements