使用python数据框将满足条件的行数列添加到满足条件的每一行

Question

假设我有一个具有以下结构的 python DataFrame：

pd.DataFrame([[1,2,3,4],[1,2,3,4],[1,3,5,6],[1,4,6,7],[1,4,6,7],[1,4,6,7]])
Out[262]: 
   0  1  2  3
0  1  2  3  4
1  1  2  3  4
2  1  3  5  6
3  1  4  6  7
4  1  4  6  7
5  1  4  6  7

如何添加一个名为“ct”的列，该列计算第 1-3 列与 匹配的每一行匹配的 DataFrame 实例...所以当所有完成了。

   0  1  2  3  ct
0  1  2  3  4  2
1  1  2  3  4  2
2  1  3  5  6  1
3  1  4  6  7  3
4  1  4  6  7  3
5  1  4  6  7  3

Answer 1

你可以使用groupby + transform + size:

df['ct'] = df.groupby([1,2,3])[1].transform('size')
#alternatively
#df['ct'] = df.groupby([1,2,3])[1].transform(len)
print (df)
   0  1  2  3  ct
0  1  2  3  4   2
1  1  2  3  4   2
2  1  3  5  6   1
3  1  4  6  7   3
4  1  4  6  7   3
5  1  4  6  7   3