如何根據先前的值設置 numpy 數組值？

Question

例子：

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

目標：

如果前一個值為 0，則將值設置為 0

所需的 output：

[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