如何从含(firm,q)多重索引的Pandas DataFrame提取q=2022q1的数据?
提取多重索引中q为"2022q1"的行方法
以下是几种针对已设置['firm', 'q']为多重索引的DataFrame,提取所有firm对应q="2022q1"行的实用方法:
方法1:使用xs()方法(推荐,简洁高效)
xs()是Pandas专门为多重索引设计的层级提取工具,直接指定目标层级和对应值即可:
df_Services.xs('2022q1', level='q')
该方法会自动匹配所有firm下的q="2022q1"行,并返回对应的子DataFrame。
方法2:使用loc索引器
通过切片匹配所有firm,再指定目标q值:
df_Services.loc[(slice(None), '2022q1'), :]
slice(None)表示匹配所有firm索引值- 逗号后的
:表示保留所有列(可按需指定列名)
方法3:使用query()方法(类SQL查询风格)
通过index.get_level_values()访问多重索引的层级,编写查询条件:
df_Services.query('index.get_level_values("q") == "2022q1"')
适合习惯用查询语句筛选数据的场景,可读性较强。
内容的提问来源于stack exchange,提问作者Dante
相关产品推荐
相关产品推荐

