numpy.split

numpy.split 함수는 어레이를 여러 개의 서브어레이로 쪼개서 리스트의 형태로 반환합니다.

axis 파라미터를 지정하면 지정한 축을 따라 어레이를 구분합니다.



예제1

import numpy as np

a = np.arange(6)

a_split1 = np.split(a, 3)
a_split2 = np.split(a, 2)

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

np.split(a, 3)은 어레이 a를 3개로 쪼갠 서브어레이를 리스트의 형태로 반환합니다.

마찬가지로 np.split(a, 2)은 어레이 a를 2개로 쪼갠 서브어레이를 리스트의 형태로 반환합니다.



예제2

import numpy as np

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

a_split1 = np.split(a, 2, axis=0)
a_split2 = np.split(a, 2, axis=1)

print(a_split1)
print(a_split2)
[array([[1, 2]]), array([[3, 4]])]
[array([[1],
     [3]]), array([[2],
     [4]])]

2차원 어레이 a에 대해,

np.split(a, 2, axis=0)은 어레이 a를 첫번째 축을 따라 2개의 서브어레이로 쪼개서 반환합니다.

np.split(a, 2, axis=1)은 어레이 a를 두번째 축을 따라 2개의 서브어레이로 쪼개서 반환합니다.



이전글/다음글

이전글 :
다음글 :