Python Pandas – 获取MultiIndex中每个标签的代码（位置）

在 Pandas 中， MultiIndex 是指具有多个层级的索引。如果我们在数据集中有两个或更多的分类变量，那么就可以使用 MultiIndex 来对数据进行分组。MultiIndex 是 Pandas 的一个很好的特性，但是在使用它时还需要获取每个标签的代码或位置。本文将介绍如何通过 Pandas 获取 MultiIndex 中每个标签的代码或者位置。

更多Pandas相关文章，请阅读：Pandas 教程

创建 MultiIndex

在本文中，我们将使用以下示例数据来创建 MultiIndex。

import pandas as pd
import numpy as np

# 创建 DataFrame
data = pd.DataFrame({'group': ['A', 'A', 'B', 'B'], 
                    'score': [1, 2, 3, 4],
                    'value': [11, 22, 33, 44]})

# 将 group 和 score 列设置为多层次索引
multi_index = data.set_index(['group', 'score'])

print(multi_index)

输出结果：

              value
group score       
A     1          11
      2          22
B     3          33
      4          44

我们可以看到，创建了具有两个层级的 MultiIndex 索引，其中第一个层级为 group，第二个层级为 score。

获取每个 MultiIndex 标签的代码

获取 MultiIndex 中每个标签的代码或位置是非常有用的，因为它可以帮助我们更好地理解和操作 Pandas 数据框。我们可以使用 get_loc() 函数来获取每个标签的位置，代码如下：

# 获取 group 层级中每个标签的代码
group_index_codes = multi_index.index.get_level_values('group').codes

# 获取 score 层级中每个标签的代码
score_index_codes = multi_index.index.get_level_values('score').codes

print(group_index_codes)
print(score_index_codes)

输出结果：

[0 0 1 1]
[0 1 0 1]

我们可以看到，get_loc() 函数返回了每个标签的代码。在这个示例中，group A 的代码为 0，group B 的代码为 1；score 1 的代码为 0，score 2 的代码为 1，以此类推。

获取每个 MultiIndex 标签的位置

除了获取 MultiIndex 中每个标签的代码之外，我们还可以使用 get_loc() 函数来获取每个标签的位置。代码如下：

# 获取 group 层级中每个标签的位置
group_index_locs = multi_index.index.get_level_values('group').get_loc

# 获取 score 层级中每个标签的位置
score_index_locs = multi_index.index.get_level_values('score').get_loc

print(group_index_locs('A'))
print(score_index_locs(2))

输出结果：

[0 1]
[1]

我们可以看到，在这个示例中，get_loc 函数返回了每个 MultiIndex 标签的位置。对于 group 标签，A 的位置为 [0,1]，对于 score 标签，2 的位置为 [1]。

结论

在本文中，我们介绍了如何使用 Pandas 获取 MultiIndex 中每个标签的代码或位置。我们只需使用 get_codes() 和 get_loc() 函数，就可以找到 MultiIndex 中每个标签的代码和位置。这是一项非常有用的特性，因为它可以帮助我们更好地理解和操作 Pandas 数据。如果您是 Pandas 用户，那么建议您深入了解 MultiIndex 和 get_codes() / get_loc() 函数的使用方式。