numpy.array_split

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

numpy.split 함수와 비슷하게 동작하지만 numpy.array_split()는 어레이를 균등하게 쪼개지 못하더라도

에러를 발생하지 않습니다.



예제1

import numpy as np

a = np.arange(6)

a_split1 = np.array_split(a, 3)
a_split2 = np.array_split(a, 4)

print(a_split1)
print(a_split2)

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

np.array_split(a, 3)는 어레이 [0, 1, 2, 3, 4, 5]를 세 개의 서브어레이로 쪼개고 리스트의 형태로 반환합니다.

np.array_split(a, 4)는 어레이 [0, 1, 2, 3, 4, 5]를 네 개의 서브어레이로 쪼개고 리스트의 형태로 반환합니다.

np.array_split(a, 4)가 반환하는 어레이들은 균등하지 않을 수도 있습니다.



예제2

import numpy as np

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

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

print(a_split1)
print(a_split2)

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

2차원 어레이 a에 대해서,

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

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


관련 페이지


이전글/다음글

이전글 :
다음글 :

공유하기