熊猫功能根据标签进行顺序

Question

我有一个数据，我想根据 login_status 和另一个类别值查找某个事件的序列。

user_id     content_id       login_status         
    1            2            not login               
    1            3            not login               
    1            4            login                   
    2            2            login                   
    2            3            login                   

我想制作这样的序列列

user_id     content_id       login_status         sequence
    1            2            not login               1
    1            3            not login               1
    1            4            login                   2
    2            2            login                   1
    2            3            login                   1

所以我想在用户登录时制作一个序列，当用户登录时序列是 2

这是我尝试过的代码

   df1['Seq'] = df1.groupby((df1[['user_id', 'login_status']] != df1[['user_id', 'login_status']].shift(1)).any(axis=1).cumsum()).cumcount() + 1

但是登录状态不变时的squance sum

如何解决这个问题

Answer 1

我相信你想要这样的东西，它使用 shift-cumsum 模式：

df.assign(
    sequence=df.groupby(['user_id'], sort=False)
               .apply(lambda x: (x['login_status'] != x['login_status'].shift()).cumsum())
               .values
)

每当login_status 更改时，这将增加每个user_id 的序列。