python - 将 bool 索引转换为运行的开始/结束对

Question

是否有一个 numpy 函数可以转换如下内容:

[0, 1, 0, 1, 1, 1, 0, 1, 1]

到连续范围的开始/结束对数组，如:

[[1, 2],
 [3, 6],
 [7, 9]]

Answer 1

不幸的是我没有安装 numpy，但这个逻辑应该为你完成。

import itertools
x = [0, 1, 0, 1, 1, 1, 0, 1, 1]

# get the indices where 1->0 and 0->1
diff = np.diff(x) 
diff_index = diff.nonzero(x)

# pair up the ranges using itertools
def pairwise(iterable):
    a, b = itertools.tee(iterable)
    next(b, None)
    return itertools.izip(a, b)

ranges = pairwise(x)

文档:

numpy diff
numpy nonzero itertools grouper