算法导论_P43_矩阵相乘

### 按二维数组格式进行输出二维数组
def arr2_print(arr):
    for i in range(len(arr)):
        if i == 0:
            print("[",arr[i])
        elif i == len(arr)-1:
            print(" ",arr[i],"]")
        else:
            print(" ",arr[i])
    return
### 实现矩阵相乘
def square_matrix_multiply(arr_1,arr_2):
    # 构建存储矩阵乘法结果的二维数组，并将数据初始化为 0，维度： arr_1 的行数  X  arr_2 的列数
    arr_3 = [  [0 for j in range(len(arr_1))] for i in range(len(arr_2[0])) ]
    for i in range(len(arr_1)):             # len(arr_1)    arr_1  行数
        for j in range(len(arr_2[0])):      # len(arr_2[0]) arr_2  列数
            # arr_1 的列 和 arr_2 的行 相同，为计算结果矩阵中一个元素时所需要的乘法次数
            for k in range(len(arr_1[0])):
                arr_3[i][j] = arr_3[i][j] + arr_1[i][k] * arr_2[k][j]
    return arr_3

if __name__ == '__main__':
    arr_1 = [[1,2,3],[4,5,6]]
    arr_2 = [[1,4],[2,5],[3,6]]
    arr_3 = square_matrix_multiply(arr_1,arr_2)
    # print("矩阵1：
",arr_1,"
矩阵2:
",arr_2,"
二者矩阵相乘结果为：",arr_3)
    print("矩阵1：")
    arr2_print(arr_1)
    print("矩阵2:")
    arr2_print(arr_2)
    print("二者矩阵相乘结果为：")
    arr2_print(arr_3)
''' 运行结果：
矩阵1：
[ [1, 2, 3]
  [4, 5, 6] ]
矩阵2:
[ [1, 4]
  [2, 5]
  [3, 6] ]
二者矩阵相乘结果为：
[ [14, 32]
  [32, 77] ]
'''