python - 如何在我的多索引熊猫数据框中只选择每个级别 (0) 的前两行？

Question

说我有一个df，我按两列分组。然后，我只想为按对象分组的前两行。IE

grouped_data = df.groupby(['company','person']).first() 

然后我如何为每一个选择前两行。例如，对于公司 = asda，有 8 行，即这家公司下有 9 人，但我只想要前两行。我怎样才能使用上面的数据框做到这一点？请注意，我首先使用的是因为在分组后我想逐列保留信息而不进行聚合。

标签： pythonpandaspandas-groupbymulti-index

Answer 1

如果您想为每家公司提供前两行，您可以这样做：

df.groupby('company').head(2)