python - Numpy - 如何在发生变化的索引处移动值

Question

所以我想在发生变化的一维 numpy 数组中移动我的值。配置移位样本。

input = np.array([0,0,0,0,1,0,0,0,0,0,1,1,1,0,0,1,0,0,0,0])
shiftSize = 2
out = np.magic(input, shiftSize)
print out
np.array([0,0,1,1,1,1,1,0,1,1,1,1,1,1,1,1,1,1,0,0])

例如，第一次切换发生且索引为 4，因此索引 2,3 变为“1”。下一次发生在 5，因此 6 和 7 变为“1”。

编辑:另外，没有 for 循环也很重要，因为这可能很慢(大数据集需要它)EDIT2:索引和变量名称

我尝试使用 np.diff，所以我知道发生更改的位置，然后使用 np.put，但是对于多个索引范围，这似乎是不可能的。

感谢您提前提供的帮助!

Answer 1

您想要的称为“binary dilation ”，包含在 scipy.ndimage 中:

import numpy as np
import scipy.ndimage

input = np.array([0,0,0,0,1,0,0,0,0,0,1,1,1,0,0,1,0,0,0,0], dtype=bool)
out = scipy.ndimage.morphology.binary_dilation(input, iterations=2).astype(int)
# array([0, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0])