python - 按 multiIndex 之一的最高分位数过滤数据帧行

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)])