我有一个通常在开头和结尾(有时在中间)为零的数组。
我想用在数组左侧或右侧找到的第一个值填充零。
array = [0, 0, 0, 1, 3, 5, 0, 4, 2, 0, 0, 0]
我理想地想要:
new_array = [1, 1, 1, 1, 3, 5, (5 or 4), 4, 2, 2, 2, 2]
我该怎么做?
答案 0 :(得分:0)
您可以尝试以下方法:
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]