본문 바로가기

파이썬3

리스트 나누기 (chunking, sub-list,list split)

728x90
반응형

리스트 (list)를 내가 원하는 숫자만큼으로 나눠주는 기능임.

다만, 여기서 n은 n개의 원소로 이루어진 서브리스트를 생성해줌.

def chunking_list(l,n):
    '''
    l : list
    n : how many components should be included in a sub-list?
    '''
    result=[]
    for idx in range(0,len(l),n):
        result.append(l[idx:idx+n])
    return result

위의 기능은 빠지는 것 없이 모두 채워지도록 만들었음.

예를들어 101개의 원소로 이루어진 list이고 n=5일 때는 마지막 1개가 빠져야 맞지만

위의 기능에서는 맨 마지막 sub-list에 6개가 들어가게 만듬.

 

아래의 기능이 좀 더 나은 것 같다.

numpy.array_split(ary, indices_or_sections, axis=0)
728x90
반응형