https://mingredbestcheapprice.blogspot.com/2023/07/function-of-numpy-array-with-if.html