[프로그래머스] Lv.0 /특별한 이차원 배열 1 (파이썬/Python)

1. 문제 설명

2. 풀이 과정 

범위가 n인 range(n) 원소 i와 j를 각각 반복해서 받는다.

j는 다른 리스트로 받는다.

i와 j가 같으면 1을 다르면 0을 반환한다. 

for문을 다 빠져나오면 전체 리스트를 반환한다.

이것을 한줄로 표현할 수 있다.

3. 코드

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