python - pandas groupby の後に各グループをサンプリングする

Question

これはどこかで答えられたに違いないことは知っていますが、見つけることができませんでした。

問題: groupby 操作後に各グループをサンプリングします。

import pandas as pd

df = pd.DataFrame({'a': [1,2,3,4,5,6,7],
                   'b': [1,1,1,0,0,0,0]})

grouped = df.groupby('b')

# now sample from each group, e.g., I want 30% of each group

Answer 1

ラムダを適用し、sampleparamで呼び出しますfrac:

In [2]:
df = pd.DataFrame({'a': [1,2,3,4,5,6,7],
                   'b': [1,1,1,0,0,0,0]})
​
grouped = df.groupby('b')
grouped.apply(lambda x: x.sample(frac=0.3))

Out[2]:
     a  b
b        
0 6  7  0
1 2  3  1