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

 

<내가 쓴 코드>

def solution(n):
    answer1 = []
    for i in range(n):
        answer2 = []   #여기에 넣어야 하나씩 끊긴다!
        for j in range(n):
            if i == j :
                answer2.append(1)
            else :
                answer2.append(0)
        answer1.append(answer2)
    return answer1