在布尔数组NumPy中保持前导岛和尾随岛为真

Question

假设我有一个一维纽比阵列。在

a = np.array([True True True False False False False True True False True False False False False False False True True])

实验结果：

^{pr2}$

为清晰起见编辑： 如何（1）将真值的“孤岛”保持在1d数组的前导和尾随位置，而（2）对于不在1d数组前导或尾随“孤岛”中的真值，将所有其他真值转换为False？在

我试过一个明显的非有效答案，那就是天真地重复一遍，但我还是觉得有更好的解决办法。有人有更好的主意吗？在

谢谢你的阅读！！！：）

Answer 1

方法1

这里有一种方法使用^{}和^{}来检测岛的启动和停止索引-

def fill_inner_islands(a):
    acc = np.maximum.accumulate
    start = a.argmax()
    end = a.size-a[::-1].argmax()
    a0 = ~a[start:end]
    a[start:end] = ~(acc(a0) & acc(a0[::-1])[::-1])

样本运行-

案例1：

^{pr2}$

案例2：

In [144]: a.astype(int)
Out[144]: array([1, 1, 1, 0, 0, 0, 0, 1, 1, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0])

In [145]: fill_inner_islands(a)

In [146]: a.astype(int)
Out[146]: array([1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0])

案例三：

In [148]: a.astype(int)
Out[148]: array([1, 1, 1, 0, 0, 0, 0, 1, 1, 0, 1, 0, 0, 0, 1, 1, 0, 0, 0])

In [149]: fill_inner_islands(a)

In [150]: a.astype(int)
Out[150]: array([1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0, 0])

方法2

更简单的方法-

def fill_inner_islands_v2(a):
    # Get stop and end indices of leading and trailing islands.
    # We do this by using one-off shifted slices and looking for the fall
    # in forward direction and fall in flipped direction   
    start = (a[1:] < a[:-1]).argmax()+1
    end = a.size - 1 - (a[:-1][::-1] < a[1:][::-1]).argmax()

    # Get the slice within those indices and assign as all False
    if ~a[start:end].all(): # To handle all True in input array
        a[start:end] = 0