Блог

выберите каждую первую строку каждого индекса многоиндексируемого фрейма данных pandas.

 grouped = ecommerce[["category_id", "brand", "price"]].groupby(by=["category_id", "brand"]).mean()
grouped_sort = grouped.sort_values(by=["category_id", "price"], ascending=False)
 

grouped_sort выглядит так:

теперь в этом фрейме данных я хочу выбрать в каждой категории только первые бренды с самой высокой ценой.

Кто-нибудь может мне помочь?

Следующий код может помочь:

 gsgb = grouped_sort.copy()
gsgb = gsgb.groupby(level=0)
print(type(gsgb))
gsgb.head()

for cat, df in gsgb:
    display(df.sort_values(by=["price"], ascending=False).reset_index().iloc[0])
 

Работает:
Он в основном перебирает все категории в сгруппированном фрейме данных, а затем сортирует значения на основе price , затем сбрасывает индекс и, наконец, выбирает тот, у которого максимальная цена.