Python程序添加两个矩阵

先决条件： Python中的数组，循环，列表理解程序计算两个矩阵的总和，然后在Python中打印。我们可以在Python中以各种方式执行矩阵加法。这是其中的两个。例子：

Input :
 X= [[1,2,3],
    [4 ,5,6],
    [7 ,8,9]]
 
Y = [[9,8,7],
    [6,5,4],
    [3,2,1]]
 
Output :
 result= [[10,10,10],
         [10,10,10],
         [10,10,10]]

使用嵌套循环

Python

# Program to add two matrices using nested loop
  
X = [[1,2,3],
    [4 ,5,6],
    [7 ,8,9]]
   
Y = [[9,8,7],
    [6,5,4],
    [3,2,1]]
   
  
result = [[0,0,0],
         [0,0,0],
         [0,0,0]]
  
# iterate through rows
for i in range(len(X)):
 
   # iterate through columns
   for j in range(len(X[0])):
       result[i][j] = X[i][j] + Y[i][j]
  
for r in result:
     print(r)

输出：

[10, 10, 10]
[10, 10, 10]
[10, 10, 10]

时间复杂度： O(N ² )，因为我们使用嵌套循环遍历矩阵。

辅助空间： O(N*N)，因为我们使用的是额外空间结果矩阵。

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

使用嵌套列表推导

这是使用嵌套列表推导添加两个矩阵加法的另一种方法。

Python

# Program to add two matrices
# using list comprehension
  
X = [[1,2,3],
    [4 ,5,6],
    [7 ,8,9]]
  
Y = [[9,8,7],
    [6,5,4],
    [3,2,1]]
 
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)

输出：

[10, 10, 10]
[10, 10, 10]
[10, 10, 10]

说明：-该程序的输出与上述相同。我们使用嵌套列表推导来遍历矩阵中的每个元素。

时间复杂度： O(N ² )，因为我们使用嵌套循环遍历矩阵。

辅助空间： O(N*N)，因为我们使用的是额外空间结果矩阵。