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

프로그래머스

 

문제

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

프로그래머스

 

풀이

 

정수 n이 매개변수로 주어질 때, 다음과 같은 n x n 크기의 이차원 배열을 구하는 함수를 작성하는 문제이다. 

규칙성을 찾아서 n일 때마다 원소가 1로 바뀌는 것을 주의해서 함수를 작성한다. 

 

def solution(n):
    
    answer =[]
    
    for i in range(n):
        temp = []

        element = [0] * n
        element[i] = 1
        temp.append(element)
        
        answer.extend(temp)
    
    return answer