python - 根据条件省略 pandas groupby 中的组

Question

这是我的数据框:

df = pd.DataFrame({'sym': list('aaaaaabb'), 'key': [1, 1, 1, 1, 2, 2, 3, 3], 'x': [100, 100, 90, 100, 500, 500, 700, 700]})

我按 key 和 sym 对它们进行分组:

groups = df.groupby(['key', 'sym'])

现在我想检查每个组中的所有 x 是否相等。如果它们不相等，我想从 df 中删除它。在这种情况下，我想省略第一组。这是我想要的 df:

   key sym    x
4    2   a  500
5    2   a  500
6    3   b  700
7    3   b  700

Answer 1

使用GroupBy.transform与 SeriesGroupBy.nunique并按 1 进行比较，按 boolean indexing 过滤:

df1 = df[df.groupby(['key', 'sym'])['x'].transform('nunique').eq(1)]
print (df1)

  sym  key    x
4   a    2  500
5   a    2  500
6   b    3  700
7   b    3  700