python查找数列中重复数字,查找数组中重复元素的索引(Python,NumPy)

使用np.diff和@WarrenWeckesser给出的

here方法来查找数组中的零运行：

import numpy as np

def zero_runs(a): # from link

iszero = np.concatenate(([0], np.equal(a, 0).view(np.int8), [0]))

absdiff = np.abs(np.diff(iszero))

ranges = np.where(absdiff == 1)[0].reshape(-1, 2)

return ranges

a = [34,2,3,22,22,22,22,22,22,18,90,5,-55,-19,22,6,6,6,6,6,6,6,6,23,53,1,5,-42,82]

zero_runs(np.diff(a))

Out[87]:

array([[ 3, 8],

[15, 22]], dtype=int32)

然后可以根据开始和之间的差异对其进行过滤.运行结束：

runs = zero_runs(np.diff(a))

runs[runs[:, 1]-runs[:, 0]>5] # runs of 7 or more, to illustrate filter

Out[96]: array([[15, 22]], dtype=int32)