查找连续相同字符列表的方法

输入:
[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

部分边界问题可能需要考虑,只是做个记录
利用同样的方法,应该可以找出等差数列,只不过需要再做一层。

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容