[Python] 숫자로 이루어진 사각형

<문제 설명>

정수 N의 값이 주어지면 일의자리 숫자로 이루어진 N * N 모양 사각형을 출력하는 프로그램을 작성해보세요. 이때 정수 n을 전달받아 일의 자리 숫자로 이루어진 정사각형을 출력하는 함수를 작성하고, 주어진 N을 함수로 전달하여 출력합니다.

 

# 사각형의 크기를 결정할 숫자를 입력받는다. 
N = int(input())

def print_square(N):
    num = 1
    # 0부터 N-1까지의(N회) 숫자만큼 반복문을 반복한다. (행을 n크기만큼 만드는 작업)
    for i in range(N):
    	# 0부터 N-1까지의(N회) 숫자만큼 반복문을 반복한다. (열을 n크기만큼 만드는 작업)
        for j in range(N):       
        	# num을 출력한다. end = ''은 개행없이 출력해준다. 
            print(num, end=' ')
            # num +1을 해준다. 
            num += 1
            if num > 9:
                num = 1
        print()  # 각 행을 출력한 후 줄 바꿈

print_square(N)