Jak podzielić macierz na 4 bloki za pomocą numpy?

Question

Wdrażam Mnożenie macierzy strsenów za pomocą pythona. W etapie dzielenia dzielimy większą macierz na mniejsze pod-macierze. Czy istnieje wbudowana funkcja numpy do podziału macierzy?

Answer 1

Niezupełnie, ale przy użyciu zapisu rozkładu macierzy należy to zrobić całkiem łatwo.

>>> A = np.linspace(0,24,25).reshape([5,5,]) 
>>> A 
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.]]) 

Producent B lewy górny 2x2 w A:

>>> B = A[0:2,0:2] 

Zauważ, że B jest widokiem, dzieli dane z

>>> B[1,1] = 60 
>>> print A 
[[ 0. 1. 2. 3. 4.] 
[ 5. 60. 7. 8. 9.] 
[ 10. 11. 12. 13. 14.] 
[ 15. 16. 17. 18. 19.] 
[ 20. 21. 22. 23. 24.]] 

Jeśli chcesz skopiować dane z A, użyj metody kopiowania:

>>> B = A[0:2,0:2].copy() 
>>> B 
array([[ 0., 1.], 
     [ 5., 60.]]) 
>>> B[1,1] = 600 
>>> B 
array([[ 0., 1.], 
     [ 5., 600.]]) 
>>> A 
array([[ 0., 1., 2., 3., 4.], 
     [ 5., 60., 7., 8., 9.], 
     [ 10., 11., 12., 13., 14.], 
     [ 15., 16., 17., 18., 19.], 
     [ 20., 21., 22., 23., 24.]]) 

Answer 2

Wpadłem na ten sam problem i niektóre funkcje wbudowane NumPy podzielić mi matrycę do 4 podmatryc (Kim są matryce o rozmiarze 2^N * 2^N)

tutaj to kod pisał:

upper_half = np.hsplit(np.vsplit(my_matrix, 2)[0], 2) 
lower_half = np.hsplit(np.vsplit(my_matrix, 2)[1], 2) 

upper_left = upper_half[0] 
upper_right = upper_half[1] 
lower_left = lower_half[0] 
lower_right = lower_half[1] 

Bonus do rekombinacji je za pomocą numpy:

C=np.vstack([np.hstack([c11, c12]), np.hstack([c21, c22])]) 

vsplit hsplit hstack i vstack wydają się być wykonane w tym celu