使用Python DataFrames添加了符合每行条件的行计数列



说我有一个带有以下结构的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列与匹配的每一行匹配> ...因此,当所有 ...已完成。

   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

您可以使用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

最新更新