複印通過索引

Question

我想根據定義哪些元件從a進入其中b一些規則的陣列（a）的元素複製到結果數組b。我在下面創建了一個例子。

是否有可能（通過一些智能索引？）執行一個步驟的最後一步（b[x,mask] = a[mask]）的所有x或者這只是在一個循環如圖所示來實現（在擴展方式）下面？我的真實例子中的x是相當大的（〜100），所以我認爲這可以從刪除循環中受益。

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

m1 = a > 6 
m2 = ~m1 & ((a < 8) & (a > 4)) 
m3 = ~m1 & ~m2 & (a > 1) 
m4 = ~m1 & ~m2 & ~m3 & (a < 10) 


b = np.zeros((4, 3, 3)) 
b[0, m1] = a[m1] 
b[1, m2] = a[m2] 
b[2, m3] = a[m3] 
b[3, m4] = a[m4] 

print '0\n', b[0] 
print '1\n', b[1] 
print '2\n', b[2] 
print '3\n', b[3] 

輸出：

0 
[[ 0. 0. 0.] 
[ 0. 0. 0.] 
[ 7. 8. 9.]] 
1 
[[ 0. 0. 0.] 
[ 0. 5. 6.] 
[ 0. 0. 0.]] 
2 
[[ 0. 2. 3.] 
[ 4. 0. 0.] 
[ 0. 0. 0.]] 
3 
[[ 1. 0. 0.] 
[ 0. 0. 0.] 
[ 0. 0. 0.]] 

Answer 1

我不知道這是否改善與否，但它是可以做到複製一個步驟，如果我們構建拆紗指標：

開始與一個面具清單：

In [1934]: m=[m1,m2,m3,m4] 

通過加入個人清單 - 調整爲與一個的a平或弄清1D版本：

In [1935]: ida=np.concatenate([np.ravel_multi_index(np.nonzero(z),a.shape) for z in m]) 
In [1936]: ida 
Out[1936]: array([6, 7, 8, 4, 5, 1, 2, 3, 0], dtype=int32) 
In [1937]: a.flat[ida] 
Out[1937]: array([7, 8, 9, 5, 6, 2, 3, 4, 1]) 

構建同爲3D b，考慮到不同的第一維值：

In [1938]: idb = [np.ravel_multi_index(((i,)+np.nonzero(z)),bshape) for i,z in enumerate(m)] 
In [1939]: idb 
Out[1939]: 
[array([6, 7, 8], dtype=int32), 
array([13, 14], dtype=int32), 
array([19, 20, 21], dtype=int32), 
array([27], dtype=int32)] 
In [1940]: idb=np.concatenate(idb) 

現在做副本：

In [1941]: b.flat[id1]=a.flat[ida] 
In [1942]: b 
Out[1942]: 
array([[[ 0., 0., 0.], 
     [ 0., 0., 0.], 
     [ 7., 8., 9.]], 

     [[ 0., 0., 0.], 
     [ 0., 5., 6.], 
     ... 
     [ 0., 0., 0.]]]) 

因此，副本是在一次調用中完成的，但我不得不使用兩個列表解析來構造所需的索引。