python - 在 numpy 数组中查找两个数字的指定匹配项的所有匹配项

Question

我需要实现的是获取所有索引的数组，其中在我的数据数组中填充了零和一个是从零到一的步骤。我需要非常快速的解决方案，因为我必须处理数百万个 hundrets milions 长度的数组。它将在计算中心运行。例如..

data_array = np.array([1,1,0,1,1,1,0,0,0,1,1,1,0,1,1,0])
result = [3,9,13]

Answer 1

试试这个:

In [23]: np.where(np.diff(a)==1)[0] + 1
Out[23]: array([ 3,  9, 13], dtype=int64)

100M 元素数组的时序:

In [46]: a = np.random.choice([0,1], 10**8)

In [47]: %timeit np.nonzero((a[1:] - a[:-1]) == 1)[0] + 1
1 loop, best of 3: 1.46 s per loop

In [48]: %timeit np.where(np.diff(a)==1)[0] + 1
1 loop, best of 3: 1.64 s per loop