結果 : stratified random sampling in python