如何填充特定的数组值，哪些索引与前一个和下一个非零值索引之间的距离相等？

Question

我试图只更改索引（+-1）位于其他非零值索引之间的数组的零值

Input:
a = pandas.Series(np.array[0,0,-2,0,0,0,0,0,0,0,5,0,0,0,0,0,0,0,-1,0])

Desired output:
[0,0,-2,0,0,0,999,0,0,0,5,0,0,0,999,0,0,0,-1,0]

我用一个for循环以一种非常低效的方式解决了这个问题，这个循环需要永远完成

我的目标是使用numpy.where实现，但我没有管理索引的想法

margen = 1
extra_forc = 100
for i in range(extra_forc, len(a)-extra_forc):
    if a.values[i] != 0: continue
    past = np.nonzero(a.values[i-extra_forc:i][::-1])[0]
    fut = np.nonzero(a.values[i+1:i+1+extra_forc])[0]
    if len(past) > 0 and len(fut) > 0:
        if abs(np.amin(past) - np.amin(fut)) <= margen:
            a.values[i] = 999

这就是我现在尝试的，但没有成功：

a = np.where((abs(np.amin(np.nonzero([a.shift(-i) for i in range(extra_forc)])[0], axis=0) - np.amin(np.nonzero([a.shift(i) for i in range(extra_forc)][::-1])[0], axis=0)) < margen), 999, a)

Answer 1

受到启发，找到了更好的解决方案。可能不是最好的，但效果相当不错

non_cero_idxs = np.nonzero(a)[0]
ceros_idxs = np.convolve(non_cero_idxs, [0.5, 0.5], mode='same').astype(dtype=np.int32)
a.loc[ceros_idxs[1:]] = 999