[python 문법] 리스트 쪼개기 및 for문을 이용한 combinations
백준 문제를 풀다가, 리스트를 내가 원하는 갯수대로 쪼개는 방법을 알고 싶어 이것저것 해봤습니다. 1. for문과 append로 만들기 arr = [1, 2, 3, 4, 5, 6] n = 2 div_li = [] for i in range(0, len(arr), n) : div_li.append(arr[i:i+n]) print(div_li) ''' (출력) [[1, 2], [3, 4], [5, 6]] ''' 2. def 함수에서 yield로 만들기 arr = [1, 2, 3, 4, 5, 6] def div_li(l, n) : for i in range(0, len(l), n) : yield l[i:i+n] n = 2 result = list(div_li(arr, n)) print(result) '''(출..
2022. 1. 16.