python - 从列 python 中计算最新的相同值

Question

我有这样的数据:

   id   a   b   c   d
    1   y   y   z   z
    2   y   z   y   y
    3   y   y   y   y

我想从上次更改的 "y" 列中计算 a b c d 值，如下所示:

id    count_y
1     2
2     1
3     4

你能帮我解决这个问题吗？

Answer 1

你可以在 cumprod 之后做 set_index

s = df.set_index('id').eq('y').cumprod(axis = 1).sum(axis = 1)
s = s.reset_index(name='count_y')
s
Out[33]: 
   id  count_y
0   1        2
1   2        1
2   3        4