Python - 替换数组中 1 之前的值

假设我有一个由 0 和 1 组成的 pandas 系列，但这可以与 numpy 数组或任何可迭代对象一起使用。我想创建一个公式，它接受一个数组和一个输入 n，然后返回一个新系列，其中第 n 个索引处包含 1，直到每次原始系列中至少有一个 1 时。这是一个例子：

array = np.array([0, 0, 0, 1, 1, 0, 0, 0, 0, 1, 0, 0, 1, 0, 0, 0, 1, 1, 1, 1, 0, 0, 1])

> preceding_indices_function(array, 2)
np.array([0, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1])

每次输入数组中存在 1 时，无论原始数组中该索引中是否有 0 或 1，它前面的两个索引都将填充为 1。

我真的很感激对此的一些帮助。谢谢！

使用卷积np.convolve https://numpy.org/doc/stable/reference/generated/numpy.convolve.html:

N = 2

# craft a custom kernel
kernel = np.ones(2*N+1)
kernel[-N:] = 0
# array([1, 1, 1, 0, 0])

out = (np.convolve(array, kernel, mode='same') != 0).astype(int)

Output:

array([0, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1])