使用变换根据分组信息过滤数据帧答案

【问题标题】：Using transform to filter dataframe based on groupby information使用变换根据分组信息过滤数据帧
【发布时间】：2022-10-20 15:35:07
【问题描述】：

我想过滤掉在下面的数据集中没有出现 3 次的 id。我想过使用 groupby 和 transform('size')，但这不起作用。

为什么？

data = pd.DataFrame({'id':[0,0,0, 1,1,1, 2,2, 3,3,3, 4, 4],
              'info':[23,22,12,12,14,23,11,2,98,76,46,341,12]})

data[data.groupby(['id']).transform('size')==3]

【问题讨论】：

标签： python pandas

【解决方案1】：

利用：

df = data[data.groupby(['id'])['id'].transform('size')==3]

【讨论】：