Python快速替换数组中特定值的方法[重复]答案

【问题标题】：Python fast way to substitute specific values in array [duplicate]Python快速替换数组中特定值的方法[重复]
【发布时间】：2016-08-02 17:30:35
【问题描述】：

假设我有两个 numpy 数组

a = np.array([[1,2,3],[4,5,6],[7,8,9]])
b = np.array([2,7,6])

我想获得

c = np.array([[1,0,3],[4,5,0][0,8,9]])

也就是说，我想用值0 替换数组a 中其值包含在序列b 中的所有元素。最快最干净的方法是什么？（有没有类似substitute(a,b,0)）

【问题讨论】：

我没试过。我想到的唯一方法就是循环。我想知道是否有一种矢量化的方式来做到这一点。 np.select, np.choice, np.place ...据我所知都是基于条件或掩码的。

【解决方案1】：

替代答案：

c = a.copy()
c[np.in1d(a.ravel(), b).reshape(a.shape)] = 0

np.in1d 有点像“in”运算符的矢量化版本，但它仅适用于一维数组。（因此是 ravel 和 reshape 操作。）

【讨论】：

【解决方案2】：

c = a.copy()
for num in b:
    c[c == num] = 0

【讨论】：