Я знаю, что на это, должно быть, кто-то ответил, но я просто не мог найти его.
Проблема: выберите каждую группу после операции 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