[Python/파이썬] 두 행렬 더하기

두 행렬을 더하는 파이썬 프로그램을 작성해보겠습니다. 행렬의 덧셈은 같은 위치에 있는 원소끼리 더하는 것을 의미합니다.

def add_matrices(matrix1, matrix2):
    result = [[0 for _ in range(len(matrix1[0]))] for _ in range(len(matrix1))]

    # 행렬 덧셈
    for i in range(len(matrix1)):
        for j in range(len(matrix1[0])):
            result[i][j] = matrix1[i][j] + matrix2[i][j]

    return result

# 첫 번째 행렬 입력 받기
print("첫 번째 행렬 입력:")
rows = int(input("행 수를 입력하세요: "))
cols = int(input("열 수를 입력하세요: "))

matrix1 = []
print("행렬의 원소를 입력하세요:")
for i in range(rows):
    row = []
    for j in range(cols):
        element = int(input())
        row.append(element)
    matrix1.append(row)

# 두 번째 행렬 입력 받기
print("두 번째 행렬 입력 (행 수와 열 수가 같아야 함):")
matrix2 = []
print("행렬의 원소를 입력하세요:")
for i in range(rows):
    row = []
    for j in range(cols):
        element = int(input())
        row.append(element)
    matrix2.append(row)

# 행렬 더하기
result_matrix = add_matrices(matrix1, matrix2)

# 결과 출력
print("두 행렬의 합:")
for row in result_matrix:
    print(row)

위의 프로그램은 파이썬으로 두 행렬을 더하는 기능을 제공합니다. 사용자로부터 두 행렬의 크기와 원소를 입력받고, 입력받은 행렬을 더한 결과를 출력합니다.