熊猫添加独特的计数列

时间:2015-12-24 21:18:22

标签: python-3.x pandas

给出以下数据框:

{{1}}

我希望通过lambda x函数得到每组'A'的'B'唯一值计数列('C'):

{{1}}

提前致谢!

1 个答案:

答案 0 :(得分:4)

如果PC对你的目标是正确的,也许

>>> df["C"] = df.groupby(["A","B"])["A"].transform("count")
>>> df
     A  B  C
0  foo  2  2
1  foo  4  2
2  foo  4  2
3  foo  2  2
4  bar  5  1
5  bar  4  1
6  bar  3  1
7  bar  2  1

会给你你想要的吗?我们正在对(A,B)对进行分组。

可爱的历史:这本来就是我做过的,但后来我再次尝试了,发现我不需要["A"]。但是它没有它的第二次工作的原因是我当时有C列,所以代码可以采取行动......(叹气)