python - 如果列中的值与条件匹配，则计算中位数

Question

我是 Pandas 的新手。

我的数据集:

df

A       B 
10      1
15      2  
65      3   
54      2  
51      2   
96      1  

我正在尝试添加新列 C 并计算由列 B 定义的同一组中的值的中位数。

预期结果:

df

A       B       C
10      11      53
15      2       34
65      3       65
54      2       34
51      2       34
96      1       53

我尝试过的:

df_final['C'] = df_final.groupby('B')['A'].transform('median')

我确实得到了答案，但是由于 DataFrame 很大，我不确定我的代码是否正确执行，有人可以告诉我我是否使用了正确的方法来实现这个目标吗？

Answer 1

您可以使用:

df_final['C'] = df_final.groupby('B')['A'].transform('median')

如评论中所提供。