用for循环写矩阵相乘python代码

在Python中，使用for循环实现矩阵相乘涉及到两个嵌套循环以及对元素的逐个对应相加。这是一个简单的二维数组（矩阵）相乘的例子，假设我们有两个矩阵A和B，它们的维度适合相乘（即A的列数等于B的行数）：

def matrix_multiply(A, B):
    rows_A = len(A)
    cols_A = len(A)
    cols_B = len(B)

    # 确保矩阵A的列数等于矩阵B的行数
    if cols_A != rows_B:
        print("无法进行矩阵相乘，矩阵A的列数不等于矩阵B的行数")
        return

    # 初始化结果矩阵C，其行数为A的行数，列数为B的列数
    result = [ * cols_B for _ in range(rows_A)]

    for i in range(rows_A):  # 遍历矩阵A的行
        for j in range(cols_B):  # 遍历矩阵B的列
            for k in range(cols_A):  # 遍历矩阵A的列（同时也是矩阵B的行）
                result[i][j] += A[i][k] * B[k][j]  # 计算并累加对应元素的乘积

    return result

# 示例矩阵
A = [[1, 2], [3, 4]]
B = [[5, 6], [7, 8]]

# 输出结果矩阵C
C = matrix_multiply(A, B)