矩阵相乘 (python语言)

这里把矩阵想象成二维excal表格

# 运算两个矩阵相乘的结果
def matrix_multiply(a, b):
    a_line = len(a) / len(a[0])  # a的行数(矩阵的总长度除第一行的长度)
    b_column = len(b[0])  # b的列数(第一行向量的长度)
    if a_line == b_column:  # 满足矩阵相乘条件
        for i in range(int(a_line)):  # a的行数
            for j in range(len(b[0])):  # b的列数
                for k in range(len(a)):
                    c[i][j] += a[i][k] * b[j][k]
    else:
        print('错误:不满足矩阵相乘条件\n')
        return error
    return c

时间效率为a(n^3)

猜你喜欢

转载自blog.csdn.net/zhangyu4863/article/details/80861825
今日推荐