結果 : stratified random sample in python