Создание DataFrame из Series в pandas

Как перевести результат группировки в новый датафрейм?

В результате исполнения кода:

import pandas as pd
import copy

ds = pd.read_csv('ds.csv', delimiter=',')
course_pivot = copy.deepcopy(ds.groupby(['course_title'])['student_id'].nunique())

получаю одномерный объект типа Series. При переводе course_pivot в pd.DataFrame, на выходе получается DataFrame с одним столбцом.

Как его сделать с двумя столбцами: course_title и student_id? Где student_id - это количество уникальных ID для каждого курса.

Вопрос элементарный, но я потратил пол дня, чтобы попытаться найти на него ответ.

10 строк таблицы ds


Ответы (1 шт):

Автор решения: MaxU

попробуйте так:

res = ds.groupby('course_title')['student_id'].nunique().reset_index(name="uniq_stud_id")

результат:

In [10]: res
Out[10]:
         course_title  uniq_stud_id
0  Веб-дизайн PRO 2.0             1
→ Ссылка