python pandasを使用した新group作成

Question

実現したいこと

python pandasにおいて、新しいグループを作成したいです。
groupby, cumsum, cumcountなどを使用して実現できると思いましたが、詰まってしまいました。
解決案のご教示願います。。

使用するデータ

次のようなgroupとflag(True,Falseでも可)の2つの列があります。

import pandas as pd

group = [0,0,0,1,1,2,2,2,2,2,2,3,3,3]
flag  = [0,1,1,0,0,0,1,1,0,0,1,1,0,1]

df = pd.DataFrame(list(zip(group,flag)),columns=["group","flag"])

期待するoutput

groupとflag列から、new_group列を作りたいです。
new_group列の特徴

• 同一group内でflagの値が変化するごとに、新しい値を付与
• 同一group内で複数回flag値が変化する場合、再度同じflag値をとっても別物として扱う。（下記添付図のgroup列の2をご参考ください。）

参考情報

私は下記URLサイトを参考に挑戦しましたが、わかりませんでした。
https://qiita.com/Masutani/items/3cea640da7d1f5f58af1

Answer 1

以下です。

>>> df['new_group'] = pd.DataFrame((df['group'].shift() != df['group']) | (df['flag'].shift() != df['flag'])).cumsum() -1
>>> print(df)
    group  flag  new_group
0       0     0          0
1       0     1          1
2       0     1          1
3       1     0          2
4       1     0          2
5       2     0          3
6       2     1          4
7       2     1          4
8       2     0          5
9       2     0          5
10      2     1          6
11      3     1          7
12      3     0          8
13      3     1          9

使い方の詳細は、以下をお読みください。
pandasでデータを行・列（縦・横）方向にずらすshift
pandasで累積和・累積積（cumsum, cumprod, cummax, cummin）

Comments

[June_1616]

迅速な回答ありがとうございます！
参考リンクの共有ありがとうございます。