输入:
[1,1,1,1,-1,-1,1,0,0]
输出:
[[1,1,1,1],[-1,-1],[1],[0,0]]
import numpy as np
def get_continuous_list(l):
tdiff=np.diff(l)
idx=np.concatenate([np.array([0]),np.where(tdiff!=0)[0]+1,np.array([len(l)])])
res=[l[x:y] for x,y in zip(idx[:-1],idx[1:])]
return res
部分边界问题可能需要考虑,只是做个记录
利用同样的方法,应该可以找出等差数列,只不过需要再做一层。