修改numpy數組中的值

Question

假設a和b給出如下：

a = np.arange(5)
b = [0,1,2]

我要的是，不包括在指數b ，在值a應等於-1。 因此，在上述情況下， a等於

a = array([0, 1, 2, -1, -1])

我知道一種方法，即

a[list(set(a)-set(b))] = -1

但是會花費太多時間，並且在實際編寫代碼時會導致太多復雜性。 一如既往，我一直在尋找比上述方法更好的方法。 隨意使用所需的任何工具。 另一個例子（以防萬一）：-如果

a = np.arange(12)
b = [3,5,6]

然后，我真正想要的是a = array([-1, -1, -1, 3, -1, 5, 6, -1, -1, -1, -1, -1]) PS別擔心a將始終是這樣的形式np.arange(int)和沒有的值b超過的長度a

Answer 1

>>> a = np.arange(12)
>>> b = [3,5,6]
>>> a[~np.in1d(np.arange(len(a)), b)] = -1
>>> a
array([-1, -1, -1,  3, -1,  5,  6, -1, -1, -1, -1, -1])

Answer 2

好吧，如果對象要創建與其索引相同或等於-1的值的范圍，則從所有-1開始並添加所需的數據可能會更簡單，而不是相反。

>>> a = np.full(12, -1, dtype=int)
>>> b = [3, 5, 6]
>>> a[b] = b
>>> a
array([-1, -1, -1,  3, -1,  5,  6, -1, -1, -1, -1, -1])