Numpyで配列の結合・分割

作成した配列の結合や分割もNumpyでよく利用します。

>>>import numpy as np

#2×3行列

>>>a = np.array([[1,2,3], [4,5,6]])

>>>b = np.array([[7, 8, 9], [10, 11, 12]])

#2×3行列と2×3行列を行方向に結合して4×3に

>>>np.r_[a,b]

array([[ 1, 2, 3], [ 4, 5, 6], [ 7, 8, 9], [10, 11, 12]])

#2×3行列2×3行列を列方向に結合して2×6に

>>>np.c_[a,b]

array([[ 1, 2, 3, 7, 8, 9], [ 4, 5, 6, 10, 11, 12]])

# 2×3行列の行を2分割

>>>np.vsplit(a,2)

[array(1, 2, 3), array(4, 5, 6)]

# 2×3行列の列を1列目と2列目で分割

>>>np.hsplit(a,[1,2])

[array([[1], [4]]), array([[2], [5]]), array([[3], [6]])]