在使用pandas处理multiindex dataframe时，直接通过列名访问索引级别会引发keyerror。本文将详细介绍如何利用`df.index.get_level_values()`方法，通过索引级别名称或位置，高效且准确地提取multiindex中的单列数据，避免常见的访问错误，并提供实用的代码示例。
理解Pandas MultiIndex与数据访问机制
Pandas的MultiIndex（多级索引）是一种强大的工具，用于处理具有多个层次化索引的数据。它允许我们在DataFrame中创建复杂的行标签结构，从而更好地组织和分析数据。然而，当数据列被设置为MultiIndex的一部分后，其访问方式与常规的DataFrame列有所不同。
通常，当我们想从一个标准的DataFrame中获取某一列时，可以直接使用df[‘列名’]或df.列名。但对于MultiIndex DataFrame，如果目标列是其索引的一部分，上述方法将不再适用，因为这些列已从数据区域（DataFrame的列）转移到了索引区域。尝试直接访问这些索引列时，Pandas会抛出KeyError，因为它在DataFrame的常规列中找不到匹配的键。
例如，考虑以下代码片段，其中sepal_length和sepal_width被设置为MultiIndex：
import pandas as pdfile_name = "https://raw.githubusercontent.com/uiuc-cse/data-fa14/gh-pages/data/iris.csv"df = pd.read_csv(file_name)df = df.set_index(['sepal_length','sepal_width'])print(df.head())
输出的DataFrame头部如下所示，sepal_length和sepal_width作为索引而非常规列：
petal_length petal_width speciessepal_length sepal_width 5.1 3.5 1.4 0.2 setosa4.9 3.0 1.4 0.2 setosa4.7 3.2 1.3 0.2 setosa4.6 3.1 1.5 0.2 setosa5.0 3.6 1.4 0.2 setosa
此时，如果尝试使用传统方式访问索引列，将会遇到错误：
# 尝试访问索引列，会导致KeyError# df['sepal_length'] # df.sepal_length# df.loc['sepal_length']# df.loc['sepal_length',:]
甚至直接访问df.index对象也无法通过点运算符或键索引来获取特定级别的值：
# df.index.sepal_length # AttributeError: 'MultiIndex' object has no attribute 'sepal_length'# df.index['sepal_length'] # IndexError: only integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boolean arrays are valid indices
核心解决方案：使用 index.get_level_values()
要从MultiIndex中提取单个索引级别的数据，最直接和推荐的方法是使用df.index.get_level_values()。这个方法专门设计用于从MultiIndex对象中获取指定级别（level）的所有值。
get_level_values()方法可以接受两种类型的参数来指定要提取的级别：
Poify
快手推出的专注于电商领域的AI作图工具
189 查看详情 级别名称 (Level Name)：即创建MultiIndex时使用的列名。级别位置 (Level Position)：从0开始的整数，表示该级别在MultiIndex中的位置。
方法详解与代码示例
让我们通过实际代码演示如何使用get_level_values()方法：
import pandas as pdfile_name = ("https://raw.githubusercontent.com/uiuc-cse/data-fa14/gh-pages/data/iris.csv")df = pd.read_csv(file_name)df = df.set_index(["sepal_length", "sepal_width"])# 1. 通过级别名称获取 'sepal_length' 索引级别的值sepal_length_by_name = df.index.get_level_values("sepal_length")print("通过名称获取 'sepal_length' 级别:n", sepal_length_by_name)# 2. 通过级别位置（0）获取 'sepal_length' 索引级别的值sepal_length_by_position = df.index.get_level_values(0)print("n通过位置0获取 'sepal_length' 级别:n", sepal_length_by_position)# 3. 同样，获取 'sepal_width' 级别sepal_width_by_name = df.index.get_level_values("sepal_width")print("n通过名称获取 'sepal_width' 级别:n", sepal_width_by_name)sepal_width_by_position = df.index.get_level_values(1)print("n通过位置1获取 'sepal_width' 级别:n", sepal_width_by_position)
运行结果示例：
通过名称获取 'sepal_length' 级别: Float64Index([5.1, 4.9, 4.7, 4.6, 5.0, 5.4, 4.6, 5.0, 4.4, 4.9, ... 6.7, 6.9, 5.8, 6.8, 6.7, 6.7, 6.3, 6.5, 6.2, 5.9], dtype='float64', name='sepal_length', length=150)通过位置0获取 'sepal_length' 级别: Float64Index([5.1, 4.9, 4.7, 4.6, 5.0, 5.4, 4.6, 5.0, 4.4, 4.9, ... 6.7, 6.9, 5.8, 6.8, 6.7, 6.7, 6.3, 6.5, 6.2, 5.9], dtype='float64', name='sepal_length', length=150)通过名称获取 'sepal_width' 级别: Float64Index([3.5, 3.0, 3.2, 3.1, 3.6, 3.9, 3.4, 3.4, 2.9, 3.1, ... 3.1, 3.1, 2.7, 3.2, 3.3, 3.0, 2.5, 3.0, 3.4, 3.0], dtype='float64', name='sepal_width', length=150)通过位置1获取 'sepal_width' 级别: Float64Index([3.5, 3.0, 3.2, 3.1, 3.6, 3.9, 3.4, 3.4, 2.9, 3.1, ... 3.1, 3.1, 2.7, 3.2, 3.3, 3.0, 2.5, 3.0, 3.4, 3.0], dtype='float64', name='sepal_width', length=150)
可以看到，get_level_values()成功地返回了一个Float64Index对象，其中包含了指定索引级别的所有值。这个结果可以直接用于后续的数据分析或作为新的DataFrame列。
替代方法：reset_index()
虽然get_level_values()是提取MultiIndex级别值的首选方法，但也有另一种常见做法是先将MultiIndex重置为常规列，然后再像访问普通列一样进行操作。
# 使用 reset_index() 方法sepal_length_reset = df.reset_index()['sepal_length']print("n通过 reset_index() 获取 'sepal_length' 级别:n", sepal_length_reset)
注意事项：
reset_index()会创建一个新的DataFrame，将所有索引级别转换为常规列。这种方法在某些场景下可能更直观，但如果仅仅是为了获取一个索引级别的值，它会带来额外的计算开销和内存消耗，因为它需要构建一个新的DataFrame。通常情况下，get_level_values()在性能和资源使用上更优，尤其是在处理大型数据集时。
总结与最佳实践
在Pandas MultiIndex DataFrame中，要提取单个索引级别的数据，核心方法是使用df.index.get_level_values()。
推荐方法：df.index.get_level_values(‘级别名称’) 或 df.index.get_level_values(级别位置)。这种方法直接、高效，并且不会改变原始DataFrame的结构。理解数据结构：始终记住，一旦列被设置为索引，它就不再是DataFrame的常规数据列，需要通过索引对象（df.index）来访问其值。避免常见错误：不要尝试使用df[‘列名’]或df.列名来访问MultiIndex中的级别，这会导致KeyError。
通过掌握get_level_values()方法，您可以更灵活、高效地处理Pandas MultiIndex数据结构，从而提升数据处理的效率和准确性。
以上就是Pandas MultiIndex索引列的高效提取方法的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/846486.html
微信扫一扫 
支付宝扫一扫 
