python-如何根据序列替换Numpy Ndarrays中的值

前端之家收集整理的这篇文章主要介绍了python-如何根据序列替换Numpy Ndarrays中的值 前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

我有一个带有零和一的序列的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

猜你在找的Python相关文章