[알고리즘] 배열 2차원으로 만들기

오늘의 문제

프로그래머스 - 배열 2차원으로 만들기 링크

1. 떠오르는 대로 풀어보기

answer에 for문으로 이중배열 넣어주고, for문 돌면서 맞게 넣어주었다. [ //로 몫만 구하기]
참고문헌 : [List comprehension으로 2차 배열 선언]

def solution(num_list, n):
    answer = [ [] for i in range(len(num_list)//n]
    for i in range(len(num_list)):
        answer[i//n].append(num_list[i])
    return answer

2. 다른 사람의 풀이를 보며 더 좋은 풀이는 무엇인지 탐구했다.

def solution(num_list, n):
	return [num_list[i-n:i] for i in range(n, len(num_list)+1, n)]

for문을 돌리면서 num_list를 슬라이싱한다... 이런 방법이..!
이해하기 위한 참고문헌: [Python 파이썬 for문, if문 한 줄로 쓰기] , [List Slicing]