特定のグループのみ抽出したい

Question

pandasで下のようなテーブルがある場合、noに3のあるグループ
index no
a 1
a 2
a 3
b 1
b 2
c 1
c 2
d 1
d 2
d 3

noに3のあるグループのみ抽出したいです。
index no
a 1
a 2
a 3
d 1
d 2
d 3

Answer 1

以下のような感じでできます。もっとスマートな方法があるかと思いますが。

import pandas as pd

df = pd.DataFrame({'index':['a','a','b','c','c'], 'no':[2,3,2,3,4]})
print(df)
#  index  no
#0     a   2
#1     a   3
#2     b   2
#3     c   3
#4     c   4

g = df[df['no'] == 3].groupby('index')
idx = set(g.groups.keys())
print(df[df['index'].isin(idx)])
#  index  no
#0     a   2
#1     a   3
#3     c   3
#4     c   4

Comments

[midsum0323]

解決しました！ありがとうございます。