python numpy array iterate 3X3 throughout the whole array one step at a time

Viewed 24

the following code

im = np.arange(81).reshape(9,9)
print np.swapaxes(im.reshape(3,3,3,-1),1,2)

prints out this output

[[[[ 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]]]]

But I want to print the following:

[[[[ 0  1  2]   [[ 1  2  3]   [[ 6  7  8]
   [ 9 10 11]    [10 11 12]    [11 12 13]
   [18 19 20]]   [19 20 21]]   [20 21 22]]]
etc.................................
...]]]

how do I do that?

0 Answers
Related