Lv.0 정사각형으로 만들기

문제 설명

이차원 정수 배열 arr이 매개변수로 주어집니다. arr의 행의 수가 더 많다면 열의 수가 행의 수와 같아지도록 각 행의 끝에 0을 추가하고, 열의 수가 더 많다면 행의 수가 열의 수와 같아지도록 각 열의 끝에 0을 추가한 이차원 배열을 return 하는 solution 함수를 작성해 주세요.

입출력

코드 풀이

def solution(arr):
    row = len(arr)
    col = len(arr[0])
    
    if row_len > col:
        for i in range(row):
            arr[i] += [0] * (row - col)
    elif row < col:
        for i in range(row, col):
            arr.append([0] * col)
    
    return arr

출처: 프로그래머스 코딩테스트 연습