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