https://bagelmobestcheapprice.blogspot.com/2024/07/python-numpy-where-function-with.html