[프로그래머스][PYTHON] Lv. 0 특별한 이차원 배열 1

문제설명

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

프로그래머스

 

 

 

 

내 코드

def solution(n):
    answer = [[0 for i in range(n)] for j in range(n)]
    for i in range(n):
        for j in range(n):
            if i == j:
                answer[i][j] = 1
            else:
                answer[i][j] = 0

    return answer

 

사실 처음에 2차원 배열 선언할 때, 먼저 0으로 채워야한다는 사실을 몰랐다. 

배열을 먼저 채우지 않고 값만 채워넣으면, 

이러한 오류가 발생하게 된다. 

잘 기억해야겠다!-

 

다른 사람 코드

def solution(n):
    answer=[[0]*n for i in range(n)]
    for i in range(n): answer[i][i]=1
    return answer

 

끄읏-!