主页 / 问题 / 1440377
Accepted
Vladislav Klimenko
Asked:2022-08-18 14:53:15 +0000 UTC2022-08-18 14:53:15 +0000 UTC

Python矩阵积

  • 772

给定两个整数矩阵 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

python матрицы

1 个回答

  1. Best Answer

    当乘以指示大小的矩阵时，得到一个矩阵n x p。
    所以在输出处纠正就足够了

    for i in range(n):

    并且代码适用于任何大小

    如果除了指定结果矩阵之外不使用任何 Python 技巧（因此循环在大多数语言中看起来都是一样的）：

    c = [[0]*p for _ in range(n)]
for i in range(n):
    for j in range(p):
        for k in range(m):
            c[i][j] += X[i][k] * Y[k][j]
    • 1

相关问题