AttributeError: 'numpy.ndarray' object has no attribute 'index' in Python