Python Pandas – 获取MultiIndex中的层级

在Pandas中，MultiIndex是一种特殊的数据结构，通常用于将数据按层级结构进行组织。在进行数据处理时，我们经常需要获取MultiIndex中的层级信息，本文将介绍如何使用Python Pandas获取MultiIndex中的层级。

构造MultiIndex对象

在进行演示之前，我们首先需要构造一个MultiIndex对象。Pandas提供了多种方式来构造MultiIndex对象，其中最常用的方式是使用元组（tuple）来构造。

import pandas as pd

# 使用元组构造MultiIndex对象
multi_index = pd.MultiIndex.from_tuples([
    ('A', 'cat', 'big'),
    ('A', 'dog', 'small'),
    ('B', 'cat', 'small'),
    ('B', 'dog', 'big')
])

通过上述代码，我们成功地构造了一个MultiIndex对象，该对象包含了3层索引，每层索引都有不同的取值，可以通过打印查看其具体结构。

print(multi_index)

输出结果如下：

MultiIndex([('A', 'cat', 'big'),
            ('A', 'dog', 'small'),
            ('B', 'cat', 'small'),
            ('B', 'dog',   'big')],
           )

获取层级信息

在构造好MultiIndex对象之后，我们可以通过以下代码获得其中每一层的信息：

# 获得第1层的取值
print(multi_index.get_level_values(0))

# 获得第2层的取值
print(multi_index.get_level_values(1))

# 获得第3层的取值
print(multi_index.get_level_values(2))

输出结果分别是：

Index(['A', 'A', 'B', 'B'], dtype='object')
Index(['cat', 'dog', 'cat', 'dog'], dtype='object')
Index(['big', 'small', 'small', 'big'], dtype='object')

通过上述代码，我们分别获取了MultiIndex的每一层信息，并通过打印输出了结果。