python - Pandas- 创建一个新列，其中填充另一列中的观察值

Question

我有一个 DataFrame 对象 df。 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

在 Pandas 中最快的方法是什么？

Answer 1

一个简单的 groupby+transform 就可以了:

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