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

Question

我有一个带有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'])

感谢

Answer 1

这是你的意思吗？

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