numpy、配列の異なるレイヤーの異なるスライス

Question

私は3次元配列m*n*kを持っていて、2次元のレイヤーごとにサイズi*jのサブアレイを取りたいと思っています。私はすべての層のスライスの開始の座標の2*kの配列cを持っています。私はループなしで必要なものを得るための素敵で簡単な方法はありますか？

例：

test = np.arange(18).reshape((3,3,2)) 
c = np.array([[0,1], [0, 1]]) 

test[:,:,0] = array([[ 0, 2, 4], 
        [ 6, 8, 10], 
        [12, 14, 16]]) 

test[:,:,1] = array([[ 1, 3, 5], 
        [ 7, 9, 11], 
        [13, 15, 17]]) 

私はループを持つ配列

[[[ 0, 9], 
    [ 2, 11]], 

    [[ 6, 15], 
    [ 8, 17]]] 

ソリューション取得したい：

h=2 
w=2 
layers = 2 
F = np.zeros((h,w,layers)) 
for k in range(layers): 
    F[:,:,k] = test[c[0,k]:c[0,k]+h, c[1,k]:c[1,k]+w, k] 

Answer 1

をここでベクトル化broadcastingを利用したアプローチとadvanced-indexingだ -

は、

d0,d1,d2 = np.ogrid[:h,:w,:layers] 
out = test[d0+c[0],d1+c[1],d2] 

サンプル実行 - それは与えられた入力からO/P、期待に得ることにはっきりしていないよう

In [112]: test = np.arange(200).reshape((10,10,2)) 
    ...: c = np.array([[0,1], [0, 1]]) 
    ...: 

In [113]: h=4 
    ...: w=5 
    ...: layers = 2 
    ...: F = np.zeros((h,w,layers)) 
    ...: for k in range(layers): 
    ...:  F[:,:,k] = test[c[0,k]:c[0,k]+h, c[1,k]:c[1,k]+w, k] 
    ...:  

In [114]: d0,d1,d2 = np.ogrid[:h,:w,:layers] 
    ...: out = test[d0+c[0],d1+c[1],d2] 
    ...: 

In [115]: np.allclose(F, out) 
Out[115]: True