pythonで２次元配列を等分割する方法

Question

pythonで２次元配列を指定した数で等分割する方法はありますでしょうか

たとえば

a = np.arange(36).reshape(6, 6)
print(a)
[[ 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]]

この配列の場合、４分割する場合は

print(a0)
[[0 1 2][6 7 8][12 13 14]]
print(a1)
[[3 4 5][9 10 11][15 16 17]]
print(a2)
[[18 19 20][24 25 26][30 31 32]]
print(a3)
[[21 22 23][27 28 29][33 34 35]]

このような結果が欲しいです。
また、９分割する場合は

print(a0)
[[0 1][6 7]]
print(a1)
[[2 3][8 9]]
print(a2)
[[4 5][10 11]]
print(a3)
[[12 13][18 19]]
print(a4)
[[14 15][20 21]]
print(a5)
[[16 17][22 23]]
print(a6)
[[24 25][30 31]]
print(a7)
[[26 27][32 33]]
print(a8)
[[28 29][34 35]]

このような結果が欲しいです。
取り急ぎ等分割ができるケースのみで大丈夫です。
ご教示頂ければ幸いです。よろしくお願い致します。

Answer 1

import numpy as np
sq = np.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]])

sub = 3  # 分割後の辺の長さ。　3ならば[3x3]の4つの正方行列に分割。　

a, b = sq.shape
ret = sq.reshape(a//sub, sub, b//sub, sub).swapaxes(1, 2)

print(ret)

[[[[ 0  1  2]
   [ 6  7  8]
   [12 13 14]]

  [[ 3  4  5]
   [ 9 10 11]
   [15 16 17]]]


 [[[18 19 20]
   [24 25 26]
   [30 31 32]]

  [[21 22 23]
   [27 28 29]
   [33 34 35]]]]

Comments

[raplus]

ご回答ありがとうございました。参考にさせて頂きました。大変助かりました。

Answer 2

難しい道具を使わないのであれば、以下です。

a = np.arange(36).reshape(6, 6)
n = 3
result = []
p, q = a.shape
for x in range(0, p, p//n):
    for y in range(0,q, q//n):
        result.append(a[x:x+p//n, y:y+q//n])

print(result)

変数a0,・・・,a8とかに代入したいのであれば、以下になります。

a = np.arange(36).reshape(6, 6)
n = 3
p, q = a.shape
for i, x in enumerate(range(0, p, p//n)):
    for j, y in enumerate(range(0,q, q//n)):
        globals()[f'a{i*n+j}'] = a[x:x+p//n, y:y+q//n]

このあと表示すると

>>> print(a0)
[[0 1]
 [6 7]]
>>> print(a1)
[[2 3]
 [8 9]]
>>> print(a2)
[[ 4  5]
 [10 11]]
>>> print(a3)
[[12 13]
 [18 19]]
>>> print(a4)
[[14 15]
 [20 21]]
>>> print(a5)
[[16 17]
 [22 23]]
>>> print(a6)
[[24 25]
 [30 31]]
>>> print(a7)
[[26 27]
 [32 33]]
>>> print(a8)
[[28 29]
 [34 35]]

Comments

[raplus]

欲しい結果が得られました。ありがとうございました。
コードが理解しやすかったのでベストアンサーとさせて頂きます。

Answer 3

結果が違っていたので書き直しました。

import numpy as np 

div = 4#分割したい数
sqrt_div = int(np.sqrt(div))
a = np.arange(36).reshape(6,6)
a = np.split(np.vstack(np.hsplit(np.hstack(np.hsplit(a, sqrt_div)), sqrt_div)),div)
print(np.msort(a))

[[[ 0  1  2]
  [ 6  7  8]
  [12 13 14]]

 [[ 3  4  5]
  [ 9 10 11]
  [15 16 17]]

 [[18 19 20]
  [24 25 26]
  [30 31 32]]

 [[21 22 23]
  [27 28 29]
  [33 34 35]]]

Comments

[ppin]

書き直しましたー！

[raplus]

ご回答ありがとうございました。参考にさせて頂きました。大変助かりました。

Answer 4

もっと簡単な方法があるかもしれませんが、とりあえず。

import numpy as np


def vs_split(arr, sec):
    return [
        block
        for rows  in np.vsplit(arr,  sec)
        for block in np.hsplit(rows, sec)
    ]

src = np.arange(36).reshape(6, 6)

print('-' * 42)
dst = vs_split(src, sec=2)
print(*dst, sep='\n')

print('-' * 42)
dst = vs_split(src, sec=3)
print(*dst, sep='\n')

Comments

[raplus]

ご回答ありがとうございました。参考にさせて頂きました。大変助かりました。