python - Pandas- 別の列の観測数で満たされた新しい列を作成します

Question

私は DataFrame object を持っていますdf。の列の値の 1 つはdf、ID 同じ ID の行が多数あります。

num_totals各 ID の観測数をカウントする新しい列を作成したいと考えています。たとえば、次のようなものです。

ID | Num Totals
1  |    3
1  |    3
1  |    3
2  |    2
2  |    2
3  |    3
3  |    3
3  |    3
4  |    1

パンダでこれを行う最速の方法は何ですか?

Answer 1

単純な groupby+transform が機能します。

df['num_totals'] = df.groupby('ID').transform('count')