Pandas でgroupby によって、含まれる値 データ抽出

Question

id	name	state
1	A	yes
2	B	yes
3	B	no
4	C	yes
5	C	yes
6	D	no

上記のDataframeがあった場合、df.groupby(['name']) して、stateに yes が含まれている場合の name を抽出するには、どのような方法があるのでしょうか？

上記の例では、A,B,C が抽出対象となります。

以上、アドバイスをお願い致します。

Answer 1

df.groupby('name').apply(lambda d:(d['state']=='yes').any())

で良いのではないでしょうか。

import pandas as pd
df = pd.DataFrame({
    'id':[1,2,3,4,5,6],
    'name':['A','B','B','C','C','D'],
    'state':['yes','yes','no','yes','yes','no']})

ret = df.groupby('name').apply(lambda d:(d['state']=='yes').any())
#name
#A     True
#B     True
#C     True
#D    False
#dtype: bool

print(ret[ret].index.values)
#['A' 'B' 'C']