pandas MultiIndex.swaplevel() 交换多层索引层级

pandas.MultiIndex.swaplevel() 方法用于交换 MultiIndex 中的两个层次（索引级别），并返回一个新的 MultiIndex 对象。默认情况下，它交换最后两个层次。此方法适用于需要重新排列多层索引顺序的场景。

语法

签名

MultiIndex.swaplevel(i=-2, j=-1)

参数说明

• i：
  可选参数。指定要交换的第一个层次的索引号或名称。

  • 类型：int 或 str，default -2
  • 含义：表示要交换的第一个层次。可以使用整数指定层次的位置，也可以使用字符串指定层次的名称。默认值为倒数第二个层次（-2）。

• j：
  可选参数。指定要交换的第二个层次的索引号或名称。

  • 类型：int 或 str，default -1
  • 含义：表示要交换的第二个层次。可以使用整数指定层次的位置，也可以使用字符串指定层次的名称。默认值为最后一个层次（-1）。

参数总结

返回值

• 返回值：
  返回一个新的 MultiIndex 对象，交换了指定层次的顺序。

使用场景

swaplevel() 方法适用于需要改变 MultiIndex 各层次顺序的场景。例如，当数据分析的需求变化时，可以通过交换层次，调整 MultiIndex 结构以适应新的分析要求。

示例

示例 1：交换最后两个层次

import pandas as pd

# 创建一个三层的 MultiIndex
multi_index = pd.MultiIndex.from_tuples(
    [('a', 1, 'x'), ('a', 2, 'y'), ('b', 1, 'z'), ('b', 2, 'w')], 
    names=['letter', 'number', 'code']
)

# 交换 'number' 和 'code' 层（默认交换最后两层）
swapped_index = multi_index.swaplevel()

# 输出交换后的 MultiIndex
f"Swapped 'number' and 'code':\n{swapped_index}\n"

输出:

Swapped 'number' and 'code':
MultiIndex([('a', 'x', 1),
            ('a', 'y', 2),
            ('b', 'z', 1),
            ('b', 'w', 2)],
           names=['letter', 'code', 'number'])

在此示例中，默认交换了 MultiIndex 中的最后两层（number 和 code），生成了一个新的 MultiIndex。

示例 2：指定层次交换

# 指定交换 'letter' 和 'code' 层
swapped_index_specific = multi_index.swaplevel('letter', 'code')

# 输出交换后的 MultiIndex
f"Swapped 'letter' and 'code':\n{swapped_index_specific}\n"

输出:

Swapped 'letter' and 'code':
MultiIndex([('x', 1, 'a'),
            ('y', 2, 'a'),
            ('z', 1, 'b'),
            ('w', 2, 'b')],
           names=['code', 'number', 'letter'])

在这个例子中，明确指定交换 letter 和 code 层次，生成了新的 MultiIndex，其中层次顺序发生了变化。

总结

MultiIndex.swaplevel() 方法非常灵活，可以帮助用户在处理多层次索引时，根据需要调整层次的顺序。无论是默认交换最后两层，还是指定任意两个层次进行交换，都能够方便地通过此方法实现。

参考

• https://pandas.pydata.org/docs/reference/api/pandas.MultiIndex.swaplevel.html

更新时间：2024-10-24 08:24:04 标签：pandas python 多层索引