[python] NumPy에서 배열을 분할하는 방법은 무엇인가요?

numpy.split() 함수는 배열을 지정된 경계로 분할합니다. 해당 함수는 분할된 배열을 반환하는데, 이때 반환되는 값은 여러 개의 하위 배열로 구성된 Python 리스트입니다.

다음은 numpy.split() 함수를 사용하여 배열을 분할하는 예제 코드입니다.

import numpy as np

# 분할할 배열 생성
arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])

# 배열을 3개로 분할
sub_arrays = np.split(arr, 3)

# 분할된 배열 출력
for sub_array in sub_arrays:
    print(sub_array)

위의 코드에서는 numpy.array() 함수를 사용하여 분할할 배열을 생성합니다. 그런 다음 numpy.split() 함수에 분할할 배열과 분할할 개수를 전달합니다.

이 예제에서는 배열 arr을 3개로 분할하였으므로, numpy.split() 함수는 3개의 하위 배열로 구성된 리스트를 반환합니다. 각 하위 배열은 원래 배열의 요소를 지정된 개수로 동등하게 분할한 결과입니다.

분할된 배열을 사용하고자 할 때는 각 하위 배열에 대해 반복문을 사용하여 접근할 수 있습니다. 예제 코드에서는 각 하위 배열을 출력하는 방법을 보여주고 있습니다.

여러 가지 다른 방법으로 배열을 분할할 수 있으므로, numpy.split() 함수 외에도 numpy.array_split() 함수 등을 사용하여 원하는 방식으로 배열을 분할할 수 있습니다.

참고 문서: