https://mobhitsbestcheapprice.blogspot.com/2023/01/python-11-stratified-sampling-per-each.html