熊猫：日期之间的计数值差值

Question

我有数据帧

ID date 
111 11-11-2016 
111 14-11-2016 
111 17-11-2016 
222 24-11-2016 
222 27-11-2016 

我需要统计数据之间的差异，以每一个ID。 我使用

df['duration'] = df.groupby(['ID','date']).date.apply(lambda x: x - x.iloc[0]) 
idx = df.groupby(['ID'])['duration'].transform(max) == df['count date'] 

但它返回错误的结果。 我如何获得理想？ 我需要

ID count date 
111 6 
222 3 

Answer 1

>>> df.groupby('ID')['date'].apply(lambda x: x.max() - x.min()) 
ID 
111 6 days 
222 3 days 

也有可能与numpy.ptp

>>> df.groupby('ID')['date'].agg(np.ptp) 
ID 
111 6 days 
222 3 days 

Answer 2

使用内置minagg的和max
See this post

d1 = df.groupby('ID').date.agg(['min', 'max']).diff(axis=1)['max'] 

ID 
111 6 days 
222 3 days 
dtype: timedelta64[ns]