Создать колонку в пандах из агрегирования

Question

У меня есть dataframe 'xyz', и я хочу создать новый столбец на основе простого вычисления, но когда я запустил код ниже, результатом будет NaN.

хуг =

account_id date  
0 123  2016-01-01 
1 124  2016-01-01 
2 125  2016-01-01 
3 126  2016-01-01 
4 123  2016-01-02 
5 124  2016-01-02 
6 125  2016-01-02 
7 126  2016-01-02 

Новая колонка Я хочу создать: количество дней, когда у меня есть данные за ACCOUNT_ID.

код Я выполнения:

xyz['new_column'] = xyz.groupby('account_id').date.nunique() 

Результат я получаю:

account_id date   new_column 
0 123  2016-01-01  NaN 
1 124  2016-01-01  NaN 
2 125  2016-01-01  NaN 
3 126  2016-01-01  NaN 
4 123  2016-01-02  NaN 
5 124  2016-01-02  NaN 
6 125  2016-01-02  NaN 
7 126  2016-01-02  NaN 

Спасибо заранее!

Answer 1

Вы можете использовать transform:

xyz['new_column'] = xyz.groupby('account_id').date.transform('nunique') 

Answer 2

Вот альтернативное решение:

xyz['new_column'] = xyz.date.map(dict(xyz.date.value_counts()))