Pandas - subtrahieren min Datum von max Datum für jede Gruppe

Question

ich eine Spalte hinzufügen möchten, die ein Ergebnis der Subtraktion von min Datum von max Datum für jede customer_id dieser Tabelle ist

Eingang:

action_date customer_id 
2017-08-15  1 
2017-08-21  1 
2017-08-21  1 
2017-09-02  1 
2017-08-28  2 
2017-09-29  2 
2017-10-15  3 
2017-10-30  3 
2017-12-05  3 

Und bekommen diese Tabelle

Ausgang:

action_date customer_id diff 
2017-08-15  1   18 
2017-08-21  1   18 
2017-08-21  1   18 
2017-09-02  1   18 
2017-08-28  2   32 
2017-09-29  2   32 
2017-10-15  3   51 
2017-10-30  3   51 
2017-12-05  3   51 

habe ich versucht, diesen Code, aber es bringt eine Menge Nans

group = df.groupby(by='customer_id') 
df['diff'] = (group['action_date'].max() - group['action_date'].min()).dt.days 

Answer 1

Sie transform Methode verwenden:

In [23]: df['diff'] = df.groupby('customer_id') \ 
         ['action_date'] \ 
         .transform(lambda x: (x.max()-x.min()).days) 

In [24]: df 
Out[24]: 
    action_date customer_id diff 
0 2017-08-15   1 18 
1 2017-08-21   1 18 
2 2017-08-21   1 18 
3 2017-09-02   1 18 
4 2017-08-28   2 32 
5 2017-09-29   2 32 
6 2017-10-15   3 51 
7 2017-10-30   3 51 
8 2017-12-05   3 51