Преобразование вывода groupby в фрейм данных - PullRequest
0 голосов
/ 09 мая 2020

Я работаю с базой данных Northwind ( ссылка для скачивания , документация ).

Хотел проверить популярность проданных продукты, я сделал это следующим образом

order_details_products = order_details.merge(products, on = 'productID')
order_details_products_categories = order_details_products.merge(categories, on = 'categoryID')

popularity_products = order_details_products_categories.groupby(['productID', 
                                                                 'categoryID', 
                                                                 'categoryName']).agg(no_orders 
                                                                                      = ('productID', 
                                                                                         'count'))
popularity_products.sort_values(by = 'no_orders', ascending = False, inplace = True)

Работает и отлично возвращает то, что мне нужно. Однако есть проблема, с которой я не знаю, как решить. Когда я проверяю форму своего фрейма данных популярные_продукты , я получаю следующий результат:

popularity_products.shape
Out[27]: (77, 1)

Это означает, что мой фрейм данных содержит только один столбец: no_orders . Можно ли создать фрейм данных с 4 столбцами: productID, categoryID, categoryName ?

С уважением.

1 Ответ

1 голос
/ 09 мая 2020

Должна быть возможность использовать:

popularity_products = popularity_products.to_frame().reset_index()
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...