[Python] 프로그래머스 lv.0 특이한 이차원 배열 1

 

정답 코드

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

 

다른 사람 코드 1

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