Python: Split NumPy array based on values in the array,"np.diff(arr[:, (1)])"