프로그래머스 0. 2차원으로 만들기 (파이썬)

 

<풀이>

주어진 num_list를 n개씩 잘라서 answer 배열에 넣으면 된다. 반복문을 통해 num_list를 n개로 나눈 몫 만큼 배열에 append 하여 문제를 풀었다. range 함수의 범위는 num_list의 길이를 n으로 나눈 몫으로 설정했다. append 될 범위는 num_list의 i*n:n*(i+1) 까지이다.

 

def solution(num_list, n):
    answer = []
    
    #n개씩 잘라서 배열에 넣기 -> for문
    #n, n+1 씩 append -> *i 해서 반복문 돌아가는 만큼 들어가도록
    #반복문 돌아가는 횟수 = numlist를 n으로 나눈 몫
    for i in range(len(num_list)//n):
        answer.append(num_list[i*n : n*(i+1)])
    return answer