他の配列の別の配列のすべての出現の同じ位置に与えられた数字をどのように置くか？

Question

別の配列の別の配列のすべての出現の同じ位置に与えられた数を入れたいと思います。

私が言っていることを理解するには、私が望むことをする単純な機能を作って、あなたに実行を示しています。しかし、

私の質問です：すでにnumpyのようなモジュールにコマンドがありますか？

機能

def swap(tab1,tab2): 
    n, m = tab1.shape 
    for i in range(n): 
     for j in range(m): 
      if tab1[i,j]==1.: 
       tab2[i,j]=2. 
    return tab2 

実行

tab1 = np.array([[0,1.,0],[1.,1.,1.],[0,1.,0]]) 
tab2 = np.array([[1.,1.,0],[1.,0.,1.],[0,1.,1.]]) 

tab1 

array([[ 0., 1., 0.], 
     [ 1., 1., 1.], 
     [ 0., 1., 0.]]) 

tab2 

array([[ 1., 1., 0.], 
     [ 1., 0., 1.], 
     [ 0., 1., 1.]]) 

saw(tab1,tab2) 

array([[ 1., 2., 0.], 
     [ 2., 2., 2.], 
     [ 0., 2., 1.]]) 

Answer 1

numpyのは、この種のものではかなり良いです。 np.whereと空想的な索引付けは次のように使用できます。

tab2[np.where(tab1==1.)] = 2. 

---

デモンストレーション：

>>> import numpy as np 
>>> tab1 = np.array([[0,1.,0],[1.,1.,1.],[0,1.,0]]) 
>>> tab2 = np.array([[1.,1.,0],[1.,0.,1.],[0,1.,1.]]) 
>>> tab2[np.where(tab1==1.)] = 2.0 
>>> tab2 
array([[ 1., 2., 0.], 
     [ 2., 2., 2.], 
     [ 0., 2., 1.]])