解答例_3_8

練習問題_3_8（forの３重化 行列の積）

def mult_matrix(A, B):
    """行列の積
    
    引数　A, B は共に行列を表す2次元リストAの列数とBの行数は同じ
    戻り値　　AとBの積行列を表す2次元リスト
    """
    AB = []
    for i in range(len(A)):
        row = []
        for j in range(len(B[0])):
            x = 0
            for k in range(len(A[0])):
                x += A[i][k] * B[k][j]
            row.append(x)
        AB.append(row)
    return AB

A, B = create_2d_list(3, 4), create_2d_list(4, 4)
mult_matrix(A, B)

[[140, 146, 152, 158], [740, 786, 832, 878], [1340, 1426, 1512, 1598]]