[programmers - python] Lv.0: 2차원으로 만들기

문제 링크

프로그래머스

문제 풀이

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

2차원으로 만들기 입출력 테스트

*key point: 반복문와 리스트를 활용한다.

 

*comment: list comprehension으로 훨씬 간단한 풀이가 가능하다.

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