给定两个整数矩阵 A 和 B。矩阵 A 由 N 行和 M 列组成,矩阵 B 由 M 行和 P 列组成。需要计算这些矩阵 A B 的乘积 输入 输入文件 INPUT.TXT 的第一行包含三个自然数 N、M 和 P。下面是矩阵 A 和 B 的描述。矩阵 A 由 N 行每个 M 个整数。矩阵 B 由 M 行 P 个数组成。矩阵由空行相互分隔。输入中所有数字的绝对值不超过 100。输出 在输出文件 OUTPUT.TXT 中输出 A B乘积的矩阵。
我的代码:
n, m, p = map(int, input().split())
X = [list(map(int, input().split())) for i in range(n)]
input()
Y = [list(map(int, input().split())) for j in range(m)]
result = [[sum(a * b for a, b in zip(X_row, Y_col)) for Y_col in zip(*Y)] for X_row in X]
for i in range(m):
for j in range(p):
print(result[i][j], end=' ')
print()
输入测试数据:维度 2 2 3
矩阵 A
2 3
-十四
矩阵 B
2 -3 4
3 1 0
结论:
13 -3 8
10 7 -4
我找不到不同大小的矩阵的解决方案,2x2 x 2x3