generator를 사용하여 넘파이 배열을 일정한 크기씩 나누어 가져오기

generator를 사용하여 넘파이 배열을 일정한 크기씩 나누어 가져오는 것을 작성한 테스트 코드입니다.

더 간단한 방법이 있을 거 같았는데 마땅히 떠오르지 않습니다.

 

2021. 11. 28  최초작성

 

import numpy as np


a = np.arange(13)
print(a)
print()

def generator(arr, n): 
    for i in range(n, len(arr) +1, n): 

        d = arr[i-n:i]

        yield d

for i,d in enumerate(generator(a, 2)):
    print(i, d)

 

실행 결과

 

[ 0  1  2  3  4  5  6  7  8  9 10 11 12]

0 [0 1]
1 [2 3]
2 [4 5]
3 [6 7]
4 [8 9]
5 [10 11]