Python Numpy Find Index Of Element Python