[파이썬] 프로그래머스_2차원으로 만들기

문제

 

https://school.programmers.co.kr/learn/courses/30/lessons/120842

프로그래머스

---

 

해설

 

n 값과 비교할 수 있는 cnt를 선언하고

num_list의 원소를 나눠 담을 수 있는 temp 배열을 선언한다.

num_list의 원소 하나씩 temp에 담고, cnt 값을 1씩 증가시켜 n과 cnt 값을 비교할 수 있도록 해준다.

cnt가 n과 같아지면 temp를 배열 answer에 넣어주고 

temp와 cnt를 초기화 시켜 모든 원소를 담을 수 있도록 반복해주면 된다.

---

코드

 

-파이썬

def solution(num_list, n):
    answer = []
    cnt = 0
    temp = []
    for i in num_list:
        temp.append(i)
        cnt += 1
        if cnt == n:
            answer.append(temp)
            temp = []
            cnt = 0

    return answer