結果 : how to do stratified sampling in python