Python Numpy Find Index Of Element In Vector