我有一个带有零和一的序列的numpy多维数组.我想将任何序列101替换为零.例如:
a = np.array([[0,1,1],[1,1]])
应该变成:
a = np.array([[0,1]])
最佳答案
我们可以使用
2D convolution
–
from scipy.signal import convolve2d
k = np.array([[1,1]]) # kernel for convolution
a[(convolve2d(a,k,'same')==2) & (a==0)] = 1