https://researchdatapod.com/python-numpy-where-function/