より大きい行列を使って部分行列を反復する

Question

forループを使ってより大きい行列を反復する方法を探しています。各反復ではサイズ（行、列、深さ）のサブ行列が出力されます（6、 3,3）。

私の大きな行列は、numpy行列として格納され、各繰り返しが出力されるようにも可能です。

>>> import numpy as np 
>>> a = np.random.rand(6*3,3*3,3) 
>>> print a.shape 
(18, 9, 3) 
>>> print a 
>>> b 

変数bは、行列aからサイズ（6,3,3）のすべてのサブ行列を含む必要があります。 各サブマトリックスは、以前のサブマトリックスと重複してはいけません。

Answer 1

アプローチ＃1

私たちはnon-overlapping/distinctブロックを探していると仮定しています。サンプル実行

from skimage.util.shape import view_as_blocks 

BSZ = (6,3,3) 
out = view_as_blocks(a,BSZ).reshape((-1,)+ (BSZ)) 

- -

In [279]: a = np.random.rand(6*3,3*3,3) 

In [280]: out = view_as_blocks(a,BSZ).reshape((-1,)+ (BSZ)) 

In [281]: out.shape 
Out[281]: (9, 6, 3, 3) 

---

reshapingとtransposeのようなだけでネイティブのnumpyのツールを使用したアプローチ＃2

が、ここでは一つの方法だなどとして、我々はScikit-image's view_as_blocksユーティリティを使用することができます -

m,n,r = a.shape 
split_shp = m//BSZ[0], BSZ[0], n//BSZ[1], BSZ[1], r//BSZ[2], BSZ[2] 
out = a.reshape(split_shp).transpose(0,2,4,1,3,5).reshape((-1,)+ (BSZ))