파이썬으로 두 행렬 곱하기

파이썬을 사용하여 두 행렬을 곱하는 프로그램을 작성해보겠습니다. 두 행렬을 곱하기 위해서는 첫 번째 행렬의 열 수와 두 번째 행렬의 행 수가 같아야 합니다.

def multiply_matrices(matrix1, matrix2):
    # 결과 행렬 초기화
    result = [[0 for _ in range(len(matrix2[0]))] for _ in range(len(matrix1))]

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

    return result

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

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

# 두 번째 행렬 입력 받기
print("두 번째 행렬 입력:")
rows2 = int(input("행 수를 입력하세요: "))
cols2 = int(input("열 수를 입력하세요: "))

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

# 두 행렬 곱하기
result_matrix = multiply_matrices(matrix1, matrix2)

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

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