python - 如何统计一列中的元素并将结果作为新列？

Question

名为df的DataFrame如下所示。

import pandas as pd 
df = pd.DataFrame({'id': [1, 1, 3]})

输入:

   id
0   1
1   1
2   3

我想统计每个id的个数，并将结果作为一个新的列count。

预期:

    id  count
0   1       2
1   1       2
2   3       1

Answer 1

pd.factorize 和 np.bincount

我的最爱。 factorize 不排序，时间复杂度为 O(n)。对于大数据集，factorize 应该优先于 np.unique

i, u = df.id.factorize()
df.assign(Count=np.bincount(i)[i])

   id  Count
0   1      2
1   1      2
2   3      1

---

np.unique 和 np.bincount

u, i = np.unique(df.id, return_inverse=True)
df.assign(Count=np.bincount(i)[i])

   id  Count
0   1      2
1   1      2
2   3      1