在pandas多重索引multiIndex中选定指定索引的行方法

作者:PIPIXIU 时间:2023-07-07 22:04:16 

在multiIndex中选定指定索引的行

我们在用pandas类似groupby来使用多重index时,有时想要对多个level中的某个index对应的行进行操作,就需要在dataframe中找到该index对应的行,在单层index中我们可以方便的使用df.loc[index]来选择,在多重Index中我们可以利用的类似的思路,然而其中也有一些小坑,记录如下。

1 index为有序的

1.1 创建测试数据

首先创建一个dataframe数据


df = pd.DataFrame({'class':['A','A','A','B','B','B','C','C'],
  'id':['a','b','c','a','b','c','a','b'],
  'value':[1,2,3,4,5,6,7,8]})

df中内容如下图:

在pandas多重索引multiIndex中选定指定索引的行方法

1.2 设置multiIndex

通过set_index设为多重索引


df = df.set_index(['class','id'])

设置索引后效果:

在pandas多重索引multiIndex中选定指定索引的行方法

1.3 切片筛选index

这里同样使用loc定位


df.loc[('A',slice(None)),:]

各参数的解释如下:

loc[(a,b),c]中第一个参数元组为索引内容,a为level0索引对应的内容,b为level1索引对应的内容

因为df是一个dataframe,所以要用c来指定列

这里‘A',指选择class中的A类

slice(None), 是Python中的切片操作,这里用来选择任意的id,要注意!不能使用‘:'来指定任意index

‘:',用来指定dataframe任意的列

执行后的结果如下:

在pandas多重索引multiIndex中选定指定索引的行方法

同样,如果想只保留id中的'a',则可以使用:


df.loc[(slice(None),'a'),:]

2 index无序

前面的例子对应的index列为数字或字母,是有序的,接下来我们看看index列为中文的情况。

2.1 创建无序测试数据


df2 = pd.DataFrame({'课程':['语文','语文','数学','数学'],'得分':['最高','最低','最高','最低'],'分值':[90,50,100,60]})
df2 = df2.set_index(['课程','得分'])

在pandas多重索引multiIndex中选定指定索引的行方法

2.2 尝试切片选择index


df2.loc[('语文',slice(None)),:]

我们进行同样的操作,这时会发现提示出错:

UnsortedIndexError: 'MultiIndex Slicing requires the index to be fully lexsorted tuple len (2), lexsort depth (0)'

这是因为此时的index无法进行排序,在pandas文档中提到:Furthermore if you try to index something that is not fully lexsorted, this can raise:

我们可以通过 df2.index.is_lexsorted()来检查index是否有序,


In[1]: df2.index.is_lexsorted()
out[1]: False

接下来,我们尝试对Index进行排序。(排序时要在level里指定index名)

2.3 对index排序后切片选择index


df2 = df2.sort_index(level='课程')

df2.loc[('语文',slice(None)),:]

在pandas多重索引multiIndex中选定指定索引的行方法

得到了我们想要的结果。

参考文献:pandas-docs-MultiIndex / Advanced Indexing

来源:https://blog.csdn.net/PIPIXIU/article/details/80232805

标签:pandas,multiIndex,索引
0
投稿

猜你喜欢

  • 基于Python写一个番茄钟小工具

    2023-05-02 07:53:37
  • 三种数据库利用SQL语句进行高效果分页

    2008-11-28 14:52:00
  • Python Pandas聚合函数的应用示例

    2022-12-13 22:39:31
  • 使用Python实现将list中的每一项的首字母大写

    2023-01-07 15:05:04
  • sqlserver 支持定位当前页,自定义排序的分页SQL(拒绝动态SQL)

    2024-01-16 03:58:06
  • python requests.post请求404问题及解决方法

    2022-08-16 09:36:23
  • 利用Python实现数值积分的方法

    2023-07-11 10:16:31
  • Goland使用Go Modules创建/管理项目的操作

    2024-04-30 10:03:02
  • 在Internet Explorer中正确使用MSXML

    2009-02-22 18:41:00
  • CSS.JS文件发布机制的思考

    2009-08-04 13:07:00
  • Mysql索引性能优化问题解决方案

    2024-01-27 12:14:41
  • linux实现定时备份mysql数据库的简单方法

    2024-01-16 07:21:57
  • X/HTML5 v.s. XHTML2(II)

    2008-06-18 13:19:00
  • python和shell监控linux服务器的详细代码

    2021-11-09 06:23:22
  • 浅谈Vue3 父子传值

    2024-05-28 15:59:21
  • Python实现判断并移除列表指定位置元素的方法

    2023-03-21 03:06:19
  • 实例详解Python装饰器与闭包

    2021-05-27 10:12:21
  • asp提高首页性能的一个技巧

    2008-04-05 06:54:00
  • 关于Keras模型可视化教程及关键问题的解决

    2021-03-19 10:39:07
  • MySQL数据库中应当如何实施info()函数

    2008-11-27 15:04:00
  • asp之家 网络编程 m.aspxhome.com