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

🎉 문제

🎉 풀이

✅ 2차원 배열을 먼저 만들고 반복문 돌려서 원소를 넣어주면 된다

✅ 2차원 배열을 선언할 때 얕은 복사를 주의해야 한다

예를 들어, 3 * 4의 2차원 배열을 만든다고 할 때

arr = [ [0] * 3 ] * 4

이렇게 선언해버리면... 모든 2차원배열의 값이 복사되는 무서운 경험을 하게 될 것이다,,,

따라서 2차원 배열을 만들고 싶다면

arr = [ [0] * 3 for i in range(4)]

이렇게 만들기를..............

 

🎉 코드

def solution(num_list, n):
    answer = [[0]*n for i in range(len(num_list)//n)]
    
    idx = 0
    for i in range(len(num_list)//n):
        for j in range(n):
            answer[i][j] = num_list[idx]
            idx += 1
    
    return answer