基于行计数设置数据帧列值

Question

我有一个名为“segments”的数据帧，看起来像这样：

               COL_A
0              10135
1              10135
2              10135
3              10140
4              10140

我想让它看起来像这样。其中Outdegree是包含列A中的值的行数，例如10135在3行中，因此每行Outdegree设置为3：

               COL_A    OUTDEGREE
0              10135    3
1              10135    3
2              10135    3
3              10140    2
4              10140    2

以下是我正在尝试但不起作用的内容：

segments['OUTDEGREE'] = len(segments[segments['COL_A'] == segments['COL_A']])

该代码将所有行相加，并将总计值提供给OUTDEGREE列

Answer 1

您可以使用len或'count'与转换函数来实现这一点

df['OUTDEGREE'] = df.groupby('COL_A')['COL_A'].transform(len)
df

或者

df['OUTDEGREE'] = df.groupby('COL_A')['COL_A'].transform('count')

出：

    COL_A   OUTDEGREE
0   10135   3
1   10135   3
2   10135   3
3   10140   2
4   10140   2