[Level.0] 정사각형으로 만들기

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

제한사항

• 1 ≤ arr의 길이 ≤ 100
• 1 ≤ arr의 원소의 길이 ≤ 100
• arr의 모든 원소의 길이는 같습니다.
• 1 ≤ arr의 원소의 원소 ≤ 1,000

문제 해결

def solution(arr):
    row = len(arr)
    col = len(arr[0])
    if row > col:
        for ar in arr:
            ar.extend([0] * (row - col))
    elif col > row:
        for _ in range(col - row):
            arr.append([0] * col)
    return arr

 

주석: 행 > 열 일때, 각 ar 배열에 접근해 .extend()로 [0]의 배열을 추가한다. 열 > 행 일때, 추가할 행 개수만큼 [0]*열 을 추가한다.