实现矩阵相乘的python函数

以下是一个简单的Python函数，用于计算两个矩阵的乘积：

def matrix_multiply(a, b):
    """
    计算两个矩阵的乘积
    :param a: 矩阵A，二维列表
    :param b: 矩阵B，二维列表
    :return: 矩阵乘积，二维列表
    """
    # 矩阵A的行数和列数
    rows_a = len(a)
    cols_a = len(a[0])
    # 矩阵B的行数和列数
    rows_b = len(b)
    cols_b = len(b[0])
    # 如果矩阵A的列数不等于矩阵B的行数，无法相乘
    if cols_a != rows_b:
        return None
    # 初始化结果矩阵
    result = [[0 for j in range(cols_b)] for i in range(rows_a)]
    # 计算矩阵乘积
    for i in range(rows_a):
        for j in range(cols_b):
            for k in range(cols_a):
                result[i][j] += a[i][k] * b[k][j]
    return result

该函数接受两个参数a和b，它们都是二维列表，表示两个矩阵。函数首先检查a和b是否满足矩阵相乘的条件，即矩阵a的列数等于矩阵b的行数，如果不满足，函数返回None。否则，函数初始化一个结果矩阵，然后使用三重循环计算矩阵乘积。最后，函数返回结果矩阵。