https://keroseneheaterreview.blogspot.com/2024/05/numpy-equivalent-of-ifelse-list.html