[프로그래머스] Lv.0 /정사각형으로 만들기 (파이썬/Python)

1. 문제 설명

2. 풀이 과정 

행row의 길이 len(arr)를 rows에 저장한다.

열col의 길이 len(arr[0])를 cols에 저장한다.

rows가 cols보다 크면 각 행을 for문으로 받아서 각 행의 끝에 차이값(rows-cols)만큼 0을 추가row.extend()한다.

만약 cols가 rows보다 크면 차이값만큼for _ in range(cols-rows) 0으로 이루어진 열[0]*cols을 추가arr.extend()한다.

반복문을 빠져나오면 arr를 반환한다.

 

3. 코드

def solution(arr):
    rows = len(arr)
    cols = len(arr[0])

    if rows > cols:
        for row in arr:
            row.extend([0] * (rows - cols))
    elif cols > rows:
        arr.extend([[0] * cols for _ in range(cols - rows)])

    return arr