Can you use groupby for your stratified sampling work?