Pandas/同じ数字が続く間ごとにカウントしたい

Question

前提・実現したいこと

Pandas dataframe内のデータのカウントを行いたいのですが、

ID	data
1	2
1	4
2	4
1	3
1	4
1	7

というようなdataframeがあったとき、IDの要素で、同じ数字が連続している間ごとにカウントを行いたいです。

つまり、カウント結果がこの例だと、

カウント結果 = [2,1,3]

のような結果を得たいです。

for文を使ってカウントし続けて、IDが変わったときに別のカウンターに切り替えるというようなやり方だと遅くて、何か賢いやり方はないでしょうか。

Answer 1

memo

np.bincount((df["ID"] != df["ID"].shift()).cumsum())[1:]

Answer 2

import pandas as pd
import io

csv_data = '''
ID,data
1,2
1,4
2,4
1,3
1,4
1,7
'''
df = pd.read_csv(io.StringIO(csv_data))

#
count = (
  df.groupby(df['ID'].ne(df['ID'].shift().bfill()).cumsum())['data']
    .count().values)

print(count)

#
[2 1 3]

Comments

[re1313]

ありがとうございます！