python - 如何根据以前的值设置numpy数组值？

Question

例子:

a = np.array([1, 0, 2, 0, 3, 4])

目标:
如果前一个值为 0，则将值设置为 0
期望的输出:

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

Answer 1

制作零位置的掩码:

m = (a == 0)

将掩码应用于数组的移位切片:

a[1:][m[:-1]] = 0

在某些情况下，您可能希望通过增加索引来移动:

i = np.flatnonzero(m[:-1]) + 1
a[i] = 0