大熊猫选择特定的基团

Question

我有一个ID列和测量列数据帧

ID measurement 
1 A 
1 B 
1 C 
1 D 
2 A 
2 B 
2 C 
2 D 
3 A 
4 A 
4 B 
4 C 
4 D 
4 E 

我想选择仅具有每ID完整测量（AD）的行和删除行其或者具有更少（对于示例ID 3）或者每个ID的更多测量（例如ID 4）。

有没有一种有效的方法来实现这一目标？

Answer 1

In [92]: df.groupby('ID').filter(lambda x: set(x['measurement']) == set('ABCD')) 
Out[92]: 
    ID measurement 
0 1   A 
1 1   B 
2 1   C 
3 1   D 
4 2   A 
5 2   B 
6 2   C 
7 2   D 

Answer 2

使用transform

In [1001]: df[df.groupby('ID')['measurement'].transform(lambda x: set(x) == set('ABCD'))] 
Out[1001]: 
    ID measurement 
0 1   A 
1 1   B 
2 1   C 
3 1   D 
4 2   A 
5 2   B 
6 2   C 
7 2   D