Fix slicing numpy array with float number