Find Element In Numpy Array Python Code