創建2D陣列上的4D視圖以將其分成固定大小

的細胞我有一個2D陣列t在numpy的：創建2D陣列上的4D視圖以將其分成固定大小

>>> t = numpy.array(range(81)).reshape((9,9)) 
>>> t 
array([[ 0, 1, 2, 3, 4, 5, 6, 7, 8], 
     [ 9, 10, 11, 12, 13, 14, 15, 16, 17], 
     [18, 19, 20, 21, 22, 23, 24, 25, 26], 
     [27, 28, 29, 30, 31, 32, 33, 34, 35], 
     [36, 37, 38, 39, 40, 41, 42, 43, 44], 
     [45, 46, 47, 48, 49, 50, 51, 52, 53], 
     [54, 55, 56, 57, 58, 59, 60, 61, 62], 
     [63, 64, 65, 66, 67, 68, 69, 70, 71], 
     [72, 73, 74, 75, 76, 77, 78, 79, 80]])

它是由兩個數字索引：行和列索引。

>>> t[2,3] 
21 
>>> t.shape 
(9, 9) 
>>> t.strides 
(72, 8)

我想要做什麼是劃分陣列劃分成固定大小的矩形單元，3×3例如。我想避免記憶複製。我儘量做到這一點的方法是創建一個視圖到與t記者形狀和進步（(3,3,3,3)和(216,24,72,8)分別）。這樣，視圖的前兩個索引就意味着3×3單元格在大網格中的位置，最後兩個將意味着元素在單元格內的位置。例如，t[0,1,:,:]將返回

array([[ 3, 4, 5], 
     [12, 13, 14], 
     [21, 22, 23]])

所以我的問題是 - 如何創建所描述的看法？我錯過了一個更簡單的方法嗎？這可以用切片語法優雅地完成嗎？

來源

2012-01-25 ulidtko

編輯：

：不需要你找出進步自己是

numpy.rollaxis(t.reshape(3, 3, 3, 3), 2, 1)

[編輯結束]

另一種方式來實現這一目標是使用numpy.lib.stride_tricks.as_strided的一種方式

>>> t = numpy.arange(81.).reshape((9,9)) 
>>> numpy.lib.stride_tricks.as_strided(t, shape=(3,3,3,3), strides=(216,24,72,8)) 
array([[[[ 0., 1., 2.], 
     [ 9., 10., 11.], 
     [ 18., 19., 20.]], 

     [[ 3., 4., 5.], 
     [ 12., 13., 14.], 
     [ 21., 22., 23.]], 

     [[ 6., 7., 8.], 
     [ 15., 16., 17.], 
     [ 24., 25., 26.]]], 


     [[[ 27., 28., 29.], 
     [ 36., 37., 38.], 
     [ 45., 46., 47.]], 

     [[ 30., 31., 32.], 
     [ 39., 40., 41.], 
     [ 48., 49., 50.]], 

     [[ 33., 34., 35.], 
     [ 42., 43., 44.], 
     [ 51., 52., 53.]]], 


     [[[ 54., 55., 56.], 
     [ 63., 64., 65.], 
     [ 72., 73., 74.]], 

     [[ 57., 58., 59.], 
     [ 66., 67., 68.], 
     [ 75., 76., 77.]], 

     [[ 60., 61., 62.], 
     [ 69., 70., 71.], 
     [ 78., 79., 80.]]]])

請注意，您提供的步幅只適用於float數組（itemsize == 8），whi樂在您的文章的例子t是int陣列（這可能會或可能不會有itemsize == 8）。

來源

2012-01-25 16:34:48

關於itemsize：是啊，你是絕對正確的。我想我會根據已經存在的數組計算所需的步幅。這樣，視圖創建例程也可以在其他2D視圖上工作。 – ulidtko

請將'rollaxis'解決方案放在答案上。 – ulidtko

你可以這樣做：

t = np.arange(81).reshape(9,9) 
t.shape = (3, 3, 3, 3) 
t = t.transpose((0, 2, 1, 3)) 

>>> print t.strides 
(108, 12, 36, 4) 

>>> print t 
[[[[ 0 1 2] 
    [ 9 10 11] 
    [18 19 20]] 

    [[ 3 4 5] 
    [12 13 14] 
    [21 22 23]] 

    [[ 6 7 8] 
    [15 16 17] 
    [24 25 26]]] 


[[[27 28 29] 
    [36 37 38] 
    [45 46 47]] 

    [[30 31 32] 
    [39 40 41] 
    [48 49 50]] 

    [[33 34 35] 
    [42 43 44] 
    [51 52 53]]] 


[[[54 55 56] 
    [63 64 65] 
    [72 73 74]] 

    [[57 58 59] 
    [66 67 68] 
    [75 76 77]] 

    [[60 61 62] 
    [69 70 71] 
    [78 79 80]]]]

轉置會返回一個觀點只要有可能，這樣你就不必擔心在知道數據類型。

來源

2012-01-25 16:36:45

創建2D陣列上的4D視圖以將其分成固定大小

回答

相關問題