题解 | #牛客网连续练习题目3天及以上的用户#

import pandas as pd

df = pd.read_csv('nowcoder.csv')

df['date'] = pd.to_datetime(df['date']).dt.date
df2 = df[['user_id','date']].drop_duplicates()

df2['rn'] = df2.sort_values('date').groupby('user_id')['date'].rank()
df2['date_group'] = df2['date'] - pd.to_timedelta(df2['rn'],unit='d')
df3 = df2.groupby('user_id')['date_group'].count()
print(df3[df3>=3])