如何使用numpy查找满足一定数目条件的连续相邻重复值及其对应的起始索引与终止索引?

假设我有a,b两个numpy.array(),a看做index数组,b看做index对应的数值,那么如何查找满足一定数量两条件的相邻重复数值的起始start_index和终止end_index。比如按照下列a,b,

a = np.array([1,2,3,4,5,6,7,8,9,10,11,12]) #index
b = np.array([1,1,4,3,3,3,5,2,2,2,2,6])    #element

我想查找b中相邻重复数值个数大于等于3(或其他数量如大于等于2)的数值以及起始和终止index,希望得到如下结果。该如何编写python,numpy程序?

c:  array([4,8])   #start_index
d:  array([6,11])  #end_index
e:  array([3,2])  
查看全部
qq_28772801
qq_28772801
2019/04/06 23:48
  • python
  • 人工智能
  • 点赞
  • 收藏
  • 回答
    私信

0个回复