NumPy：在数组中查找第一个和最后一个索引

Question

我使用带有 numpy 的 Python。

我有一个 numpy 数组 b:

b = np.array([True,True,True,False,False,True,True,False,True,True,True,True,False])

我需要找到第一个和最后一个位置，其中 b 等于 True。

对于此示例：

out_index: [0,2]
           [5,6]
           [8,11]

请问有人能建议一下，我该如何获取out_index？

- Olga

Answer 1

b = np.array([True,True,True,False,False,True,True,False,True,True,True,True,False])
idx = np.argwhere(np.diff(np.r_[False, b, False])).reshape(-1, 2)
idx[:, 1] -= 1
print idx

输出：

[[ 0  2]
 [ 5  6]
 [ 8 11]]

- HYRY