Find Element In Array Python Numpy