定义两个二维数组A、B并实现矩阵乘法其中 A=1234B=1212
二维数组A可以表示为:
A = [[1, 2], [3, 4]]
二维数组B可以表示为:
B = [[1, 2], [1, 2]]
矩阵乘法的实现如下:
def matrix_multiplication(A, B): rows_A = len(A) cols_A = len(A[0]) rows_B = len(B) cols_B = len(B[0])
# 判断A和B是否可以进行矩阵乘法
if cols_A != rows_B:
return None
# 创建一个与结果矩阵相同大小的空矩阵
C = [[0 for _ in range(cols_B)] for _ in range(rows_A)]
# 进行矩阵乘法计算
for i in range(rows_A):
for j in range(cols_B):
for k in range(cols_A):
C[i][j] += A[i][k] * B[k][j]
return C
测试矩阵乘法函数
A = [[1, 2], [3, 4]]
B = [[1, 2], [1, 2]]
result = matrix_multiplication(A, B) print(result)
原文地址: https://www.cveoy.top/t/topic/jbJb 著作权归作者所有。请勿转载和采集!