从左侧或右侧用非零值填充一维 NumPy 数组的零值？

Question

I have an array that typically has zero's at the beginning and end (and occasionally in the middle).

I want to fill the zeros with the first value found on the left or right of the array.

array = [0, 0, 0, 1, 3, 5, 0, 4, 2, 0, 0, 0]

I want ideally:

new_array = [1, 1, 1, 1, 3, 5, (5 or 4), 4, 2, 2, 2, 2]

How can I accomplish this?

Answer 1

You can try this:

import numpy as np

array = [0, 0, 0, 1, 3, 5, 0, 4, 2, 0, 0, 0]
# nonzero indices in array
nz = np.nonzero(array)[0]

# add element to a list if nonzero otherwise get its closest nonzero element
[x if x else array[nz[np.argmin(np.abs(i-nz))]] for i, x in enumerate(array)]
>>> [1, 1, 1, 1, 3, 5, 5, 4, 2, 2, 2, 2]