对在多个PANDA列上使用GROUPBY后出现的值进行计数
问题描述
我的输入数据帧如下:
使用以下代码行生成
l = [["a", 12, 12], ["a", 12, 33.], ["b", 12.3, 12.3], ["a", 13, 1]]
df = pd.DataFrame(l, columns=["a", "b", "c"])
我目前可以按如下方式累计统计频率
使用
df['freq'] = df.groupby(by=["a","b"]).cumcount()+1
它考虑了列a和列b中的公共值,并对它们进行计数。然而,我只想在列b的值不同而列a的值相同的情况下增加频率计数。下图显示了所需输出的一种形式:
如何以优化的方式在 pandas 身上实现这一点?
解决方案
可以使用双精度.groupby
。例如:
df["freq"] = df.groupby("a")["b"].apply(lambda x: x.groupby(x).ngroup() + 1)
print(df)
打印:
a b c freq
0 a 12.0 12.0 1
1 a 12.0 33.0 1
2 b 12.3 12.3 1
3 a 13.0 1.0 2
相关文章