当前位置: 代码迷 >> python >> 通过multiIndex之一的顶部quantitile过滤数据帧行
  详细解决方案

通过multiIndex之一的顶部quantitile过滤数据帧行

热度:110   发布时间:2023-06-16 14:09:27.0

我有一个带有multiIndex的数据框,我只想保留其列值在0级索引中的最高分位数的子集记录。

在下面的示例df中,我想保留其列A值在每个0级索引上位于最高75%分位数的记录。 什么是pythonic方式呢?

arrays = [np.array(['bar', 'bar', 'baz', 'baz', 'foo', 'foo', 'qux', 'qux']),
    np.array(['one', 'two', 'one', 'two', 'one', 'two', 'one', 'two'])]

df = pd.DataFrame(np.random.randn(8, 3), index=arrays, columns = ['A','B','C'])

谢谢

你是这个意思吗?

df.groupby(level=0, group_keys=False).apply(lambda g: g[g.A >= g.A.quantile(0.75)])