python: split numpy array based on values in the array,"np.diff(arr[:, (1)])"