Python: как сгруппировать кадр данных, сохраняя при этом информацию?

У меня есть кадр данных df, подобный следующему

df
    A   B
0   1  Tom
1   1  Alice
2   2  Sarah
3   3  Jack

Я хотел бы сгруппировать и рассчитывать на A и иметь что-то вроде следующего

df
    A   count  names
0   1     2    Tom, Alice
1   2     1    Sarah
2   3     1    Jack

python pandas dataframe group-by
person emax    schedule 23.08.2019    source источник
comment
@Erfan выдает ошибку TypeError: aggregate() missing 1 required positional argument: 'arg'   -  person emax    schedule 23.08.2019
comment
Это означает, что у вас нет последней версии pandas, я предлагаю вам обновить ее. В противном случае используйте df.groupby('A').agg({'A':'size', 'B': ', '.join}).rename({'A':'count'}, axis=1).reset_index()   -  person Erfan    schedule 23.08.2019