Библиотека собеса по Data Science | вопросы с собеседований

def matrix_multiply(A, B):

    # Сначала проверим, можем ли мы вообще перемножить эти матрицы

    if len(A[0]) != len(B):

        raise ValueError("Number of A columns must equal number of B rows.")

    

    # Инициализируем результирующую матрицу, заполненную нулями

    result = [[0 for _ in range(len(B[0]))] for _ in range(len(A))]

    

    # Перемножим матрицы

    for i in range(len(A)):

        for j in range(len(B[0])):

            for k in range(len(B)):

                result[i][j] += A[i][k] * B[k][j]

    

    return result



# Проверим функцию на примере

A = [[1, 2, 3],

     [4, 5, 6]]



B = [[7, 8],

     [9, 10],

     [11, 12]]



result = matrix_multiply(A, B)

for row in result:

    print(row)