python - 列值将增加并重置另外两列的更改

Question

我有数据框，我想继续增加值直到 val更改和何时 id更改重置 count值(value)

data = [['p1',1],
        ['p1',1],
        ['p1',2],
        ['p2',3],
        ['p2',5],
        ['p3',1],
        ['p3',2],
        ['p3',1]]

df = pd.DataFrame(data = data,columns = ['id','val'])

期望输出

       id val  count
    0  p1   1      1
    1  p1   1      1
    2  p1   2      2
    3  p2   3      1
    4  p2   5      2
    5  p3   1      1
    6  p3   2      2
    7  p3   1      3

我已经到了这里

df['count'] = (df.val.diff() != 0).cumsum()

这仅在 val 时改变 id 时列更改但不重置列更改

Answer 1

你可以试试groupby+transform与 lambda

df['count'] = df.groupby("id")['val'].transform(lambda x: x.ne(x.shift()).cumsum())

print(df)

   id  val  count
0  p1    1      1
1  p1    1      1
2  p1    2      2
3  p2    3      1
4  p2    5      2
5  p3    1      1
6  p3    2      2
7  p3    1      3