将组的前N行与Pandas组合成一行列表

Question

这里对Python和熊猫来说是相当陌生的

我正在尝试将基于单独列中的值的前n行组合成一行，并使用Pandas

使用一个假设的示例，假设下表已经按“金额”列对desc进行了排序：

^{tb1}$

我想生成一个表，根据store_id进行分组，每一行都是基于“amount”列的值的前n个项目的列表。因此，如果我想按store_id查看前2项，该表如下所示：

^{tb2}$

我在这里尝试了以下建议：How to combine multiple rows into a single row with pandas，但是我一直遇到一个“'GroupedData'对象不可订阅”错误

如果您能就如何解决此问题提出任何建议，我们将不胜感激。先谢谢你

Answer 1

试试这个：

(df.sort_values('amount',ascending=False)
 .groupby('store_id')
 .agg({'item':lambda x: x.iloc[0:2].tolist()}))