跳到主要内容

Python 程序:加两个矩阵

要理解这个例子,你需要了解以下Python编程主题的知识:

在Python中,我们可以将矩阵实现为嵌套列表(列表中的列表)。我们可以将每个元素视为矩阵的一行。

例如 X = [[1, 2], [4, 5], [3, 6]] 将表示一个3x2矩阵。第一行可以被选中为 X[0],第一行第一列的元素可以被选中为 X[0][0]

我们可以用Python中的不同方式进行矩阵加法。这里有几种方法。

源代码:使用嵌套循环的矩阵加法

# 使用嵌套循环加两个矩阵的程序
X = [[12,7,3],
[4 ,5,6],
[7 ,8,9]]

Y = [[5,8,1],
[6,7,3],
[4,5,9]]

result = [[0,0,0],
[0,0,0],
[0,0,0]]

# 遍历行
for i in range(len(X)):
# 遍历列
for j in range(len(X[0])):
result[i][j] = X[i][j] + Y[i][j]

for r in result:
print(r)

输出

[17, 15, 4]
[10, 12, 9]
[11, 13, 18]

在这个程序中,我们使用嵌套的for循环来遍历每一行和每一列。在每个点上,我们将两个矩阵中对应的元素相加,并将其存储在结果中。

源代码:使用嵌套列表理解的矩阵加法

# 使用列表理解加两个矩阵的程序
X = [[12,7,3],
[4 ,5,6],
[7 ,8,9]]

Y = [[5,8,1],
[6,7,3],
[4,5,9]]

result = [[X[i][j] + Y[i][j] for j in range(len(X[0]))] for i in range(len(X))]

for r in result:
print(r)

这个程序的输出与上面的相同。我们使用了嵌套列表理解来遍历矩阵中的每个元素。

列表理解允许我们编写简洁的代码,我们应该经常在Python中使用它们。它们非常有帮助。