Вот вам решение от PandasAI:
df = (df
.sort_values(by='score', ascending=False)
.groupby('book')
.head(3)
.groupby('book')['page']
.apply(list)
.reset_index()
)
Статью про PandasAI можно глянуть на хабре или в гугл колабе .
🏆 🏆 AIeron | Data Feeling @datafeeling · 14.2K subscribers 🏆 AIeron | Data Feeling 🧠 Ловите Pandas задачку на подумать ) Предлагаю решить ее двумя способами: - Самим, написав код по старинке. - С помощью любой LLM-ки Бонусом можно еще через PandasAI решить - это такой LLM агент с Pandas знаниями под капотом. Вот хороший гугл колаб с…
df = (df
.sort_values(by='score', ascending=False)
.groupby('book')
.head(3)
.groupby('book')['page']
.apply(list)
.reset_index()
)