[Numpy] concatenate, split

1. 배열 합치기 (concatenate)

기준이 되는 축을 따라 배열 순서를 결합한다.

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

np.concatenate((a, b), axis=0)  # array([[1, 2],
                                #        [3, 4],
                                #        [5, 6]])

np.concatenate((a, b.T), axis=1)  # array([[1, 2, 5],
                                  #        [3, 4, 6]])
                                  
np.concatenate((a, b), axis=None)  # array([1, 2, 3, 4, 5, 6])

 

 

 

2. 배열 나누기 (split)

배열을 여러 개의 하위 배열로 분할한다.

 

numpy.split(ary, N)

축을 따라 N등분하여 분할한다.

x = np.arange(9.0)

np.split(x, 3)  # [array([0.,  1.,  2.]), array([3.,  4.,  5.]), array([6.,  7.,  8.])]

 

 

numpy.split(ary, [a, b, ··])

축을 따라 배열이 분할되는 위치를 표시한다.
인덱스가 배열의 개수를 초과하는 경우, 비어 있는 하위 배열이 반환된다.

x = np.arange(8.0)

np.split(x, [3, 5, 6, 10])

'''
[array([0.,  1.,  2.]),
 array([3.,  4.]),
 array([5.]),
 array([6.,  7.]),
 array([], dtype=float64)]
 '''