파이썬에서 사용자로부터 행렬 입력받는 방법

이 튜토리얼에서는 파이썬에서 사용자로부터 행렬(matrix)을 입력받는 방법을 알아봅니다. 사용자 입력을 처리하는 방식은 크게 두 가지가 있으며, 각각의 방법을 예제 코드와 함께 살펴보겠습니다.

방법 1: 요소를 하나씩 입력받기

첫 번째 방법은 행렬의 숫자를 사용자에게 하나씩 입력받는 방식입니다. 중첩 반복문을 이용해 각 위치의 값을 차례대로 받아 행과 열을 구성합니다.

예제 코드

# 빈 행렬 초기화
matrix = []
# 사용자로부터 2x2 행렬 입력받기
for i in range(2):
    # 빈 행 생성
    row = []
    for j in range(2):
        # 사용자에게 숫자 입력 요청
        # 기본 입력값은 문자열이므로 int로 변환
        element = int(input())
        # 'row'에 요소 추가
        row.append(element)
    # 완성된 'row'를 'matrix'에 추가
    matrix.append(row)
# 행렬 출력
print(matrix)

실행 결과

위 코드를 실행하면 다음과 같은 결과를 얻을 수 있습니다.

1
2
3
4
[[1, 2], [3, 4]]

방법 2: 한 줄씩 공백으로 구분하여 입력받기

두 번째 방법은 한 번에 한 행씩, 공백으로 구분된 값들을 입력받는 방식입니다. map 함수와 int 함수를 조합해 입력된 문자열들을 정수로 변환합니다. 이 방법은 코드가 더 간결하고 입력 속도도 빠릅니다.

예제 코드

# 빈 행렬 초기화
matrix = []
# 사용자로부터 2x2 행렬 입력받기
for i in range(2):
    # 사용자로부터 한 행 입력받기
    row = list(map(int, input().split()))
    # 'row'를 'matrix'에 추가
    matrix.append(row)
# 행렬 출력
print(matrix)

실행 결과

위 코드를 실행하면 다음과 같은 결과를 얻을 수 있습니다.

1 2
3 4
[[1, 2], [3, 4]]

마무리

지금까지 파이썬에서 사용자로부터 행렬을 입력받는 두 가지 방법을 살펴보았습니다. 첫 번째 방법은 초보자가 행렬의 구조를 이해하기에 좋고, 두 번째 방법은 실전에서 더 효율적입니다. 상황에 맞게 활용해 보시기 바랍니다.

튜토리얼에 대해 궁금한 점이 있다면 댓글로 남겨주세요.