python 行列の分割

A = np.array([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12], [13, 14, 15, 16]])
を分割して
A1 = np.array([[1, 2],[5, 6]])
A2 = np.array([[3, 4],[7, 8]])
A3 = np.array([[9, 10],[13, 14])
A4 = np.array([[11, 12],[15, 16])

のように4分割するにはどうしたら良いのでしょうか？

行動規範の内容に同意します

回答3件

この程度のデータであれば、sliceでも十分な気がします。

Python
1import numpy as np
2
3A = np.array([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12], [13, 14, 15, 16]])
4A1 = A[0:2,0:2]
5A2 = A[0:2,2:4]
6A3 = A[2:4,0:2]
7A4 = A[2:4,2:4]
8print(A1)
9print(A2)
10print(A3)
11print(A4)

投稿2020/04/27 06:43

magichan

総合スコア15898

magichan

2020/04/27 06:44

おっと、被りました。スミマセン

行動規範の内容に同意します

Python3
1A1 = A[0:2, 0:2]
2A2 = A[0:2, 2:4]
3A3 = A[2:4, 0:2]
4A4 = A[2:4, 2:4]

投稿2020/04/27 06:41

Kenji.Noguchi

総合スコア358

ベストアンサー

numpy.hsplit() と
numpy.vsplit() を使って、以下のように分割すればよいです。

[[1, 2, 3, 4],
 [5, 6, 7, 8],
 [9, 10, 11, 12],
 [13, 14, 15, 16]]

↓ numpy.vsplit()

[[1, 2, 3, 4],
 [5, 6, 7, 8]])
-------------------------
[[ 9, 10, 11, 12],
 [13, 14, 15, 16]]

↓ numpy.hsplit()

[[1 2]   | [[3 4]
 [5 6]]  |  [7 8]]
-------------------------
[[ 9 10] | [[11 12]
 [13 14]]| [15 16]]

python
1import numpy as np
2
3A = np.array([[1, 2, 3, 4],
4              [5, 6, 7, 8],
5              [9, 10, 11, 12],
6              [13, 14, 15, 16]])
7
8blocks = []
9for cols in np.vsplit(A, 2):
10    for row in np.hsplit(cols, 2):
11        blocks.append(row)
12
13B = blocks[0]
14C = blocks[1]
15D = blocks[2]
16E = blocks[3]
17print(B)
18print(C)
19print(D)
20print(E)