https://fairytalefashionheadband.blogspot.com/2022/08/python-numpy-where-function-with.html