DataFrameから特定のIDを引っ張り出したい

Question

前提・実現したいこと

AとBの2種類のIDをDataFrameで作成しています。
AのIDに対してBのIDが複数種紐づけられる場合に、Bから紐づくIDが3個以下の場合、Aの値をリストとして集計したいです。

　A　｜　B
aaa ooo
aaa ppp
aaa qqq
aaa rrr
bbb xxx
ccc yyy
ccc zzz

この場合、条件を満たすbbbとcccを、["bbb","ccc"]で抽出するイメージです。
jupyter notebookで動かしてます。

Answer 1

import pandas as pd

df = pd.DataFrame({
    'A': ['aaa', 'aaa', 'aaa', 'aaa', 'bbb', 'ccc', 'ccc'],
    'B': ['ooo', 'ppp', 'qqq', 'rrr', 'xxx', 'yyy', 'zzz']
})

g = df.groupby('A').groups
matched_A = [k for k in g.keys() if g[k].size <= 3]