如何根據 pandas dataframe 中的多列按元素分組並將每組的元素數量保存在另一列中？

Question

我有下一個形式的 dataframe：

+--------+--------+--------+
|  Col1  |  Col2  |  Col3  |
+--------+--------+--------+
|  1     |   2    |    1   |
+--------+--------+--------+
|  1     |   2    |    1   |
+--------+--------+--------+
|  1     |   3    |    1   |
+--------+--------+--------+
|  2     |   4    |    1   |
+--------+--------+--------+ 

我想得到下一個dataframe：

+--------+--------+--------+--------+
|  Col1  |  Col2  |  Col3  |  Count |
+--------+--------+--------+--------+
|  1     |   2    |    1   |   2    |
+--------+--------+--------+--------+
|  1     |   2    |    1   |   2    |
+--------+--------+--------+--------+
|  1     |   3    |    1   |   1    |
+--------+--------+--------+--------+
|  2     |   4    |    1   |   1    |
+--------+--------+--------+--------+

我怎樣才能得到那個 dataframe？ 在按 dataframe 分組后，我嘗試使用 size() 方法，但結果不是我想要的。 我想得到另一列，其中出現整行的出現次數。

Answer 1

使用GroupBy.transform得到一個相等長度的向量，並將其作為新列：

df['Count'] = df.groupby(df.columns.tolist())['Col1'].transform('size')

Output

   Col1  Col2  Col3  Count
0     1     2     1      2
1     1     2     1      2
2     1     3     1      1
3     2     4     1      1