list_comp_matrix_multiplication_final.py

import numpy as np

def list_comp_matrix_multiplication(A, B):
    """Third and final version of the list comprehension matrix multiplication."""
    return [[sum([x*y for (x, y) in zip(row, col)]) for col in zip(*B)] for row in A]

if __name__ == '__main__':
    A = [[1, 2, 3], [4, 5, 6]]
    B = [[7, 8], [9, 10], [11, 12]]
    
    print(for_loop_matrix_multiplication(A, B))
    # The result should be: [[58, 64], [139, 154]]
    # You can check this by doing: np.matmul(A, B)