numpy.hsplit

numpy.hsplit 함수는 어레이를 수평 방향으로 여러 개의 서브어레이로 나눕니다.



예제1

import numpy as np

a = np.arange(16).reshape(4, 4)

a_hsplit = np.hsplit(a, 2)

print(a)
print(a_hsplit)

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

어레이 a는 (4, 4) 형태를 갖는 2차원 어레이입니다.

np.hsplit(a, 2)는 어레이 a를 수평 방향으로 (열 단위로) 쪼개서 리스트의 형태로 반환합니다.

numpy.hsplit 함수는 numpy.split 함수에서 axis=1으로 지정한 것과 같이 동작합니다.



예제2

import numpy as np

a = np.arange(6)

a_hsplit = np.hsplit(a, 2)

print(a)
print(a_hsplit)

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

numpy.hsplit 함수는 어레이의 차원에 관계 없이 두번째 축을 따라 어레이를 쪼개서 반환합니다.



이전글/다음글

이전글 :
다음글 :

공유하기