问题描述
我有一个带有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'])
谢谢
1楼
你是这个意思吗?
df.groupby(level=0, group_keys=False).apply(lambda g: g[g.A >= g.A.quantile(0.75)])