如何在numpy数组中找到一个零元素前面至少有N1个连续的零？

Question

给定一个numpy数组（为了简单起见，让它成为一个位数组），我如何构造一个相同形状的新数组，其中1正好位于原始数组中有一个零的位置，前面至少有N-1个连续的零？在

例如，实现具有两个参数的函数nzeros的最佳方法是什么，一个numpy数组和所需的最小连续零个数：

import numpy as np
a = np.array([0, 0, 0, 0, 1, 0, 0, 0, 1, 1])
b = nzeros(a, 3)

函数nzeros(a, 3)应返回

^{pr2}$

Answer 1

使用for循环：

def nzeros(a, n):
  #Create a numpy array of zeros of length equal to n
  b = np.zeros(n)

  #Create a numpy array of zeros of same length as array a
  c = np.zeros(len(a), dtype=int)

  for i in range(0,len(a) - n):
    if (b == a[i : i+n]).all():  #Check if array b is equal to slice in a
      c[i+n-1] = 1

  return c

样本输出：

^{pr2}$