https://camera-optical-zoom.blogspot.com/2024/02/python-11-stratified-sampling-per-each.html