仅包括一个id有2个标志的id(PANDAS)

Question

我有以下名为df的数据帧，我想将该数据帧子集为只有列is_signup有1和0的ID。下面的示例将删除id=1，因为它只有1（而不是0）。

id   tag         is_signup
1    Button      1 
1    Circle      1
2    Button      1 
2    Circle      0
2    Diamond     1 
3    Circle      0
3    Button      1 

预期产出：

id   tag         is_signup
2    Button      1 
2    Circle      0
2    Diamond     1 
3    Circle      0
3    Button      1 

我怎么能这么做？我想一个团体会有帮助？但不确定如何正式地进行

Answer 1

还可以使用groupby-filter检查is_signup.nunique():

df.groupby('id').filter(lambda x: x.is_signup.nunique() == 2)

#    id      tag  is_signup
# 2   2   Button          1
# 3   2   Circle          0
# 4   2  Diamond          1
# 5   3   Circle          0
# 6   3   Button          1

Answer 2

由于列is_signup是一个二进制值，并且只能有两个可能的值，因此我们可以在这里使用nunique来groupby+transform,然后在vaue为2（1和0）时进行布尔索引检查：

df[df['is_signup'].groupby(df['id']).transform('nunique').eq(2)]

   id      tag  is_signup
2   2   Button          1
3   2   Circle          0
4   2  Diamond          1
5   3   Circle          0
6   3   Button          1