  Python program to add two matrices

Loops | Python Methods and Functions

A program to calculate the sum of two matrices and then print it out in Python. We can perform matrix addition in various ways in Python. Here are two of them.
Examples:

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]]

Using nested

# Program for adding two matrices using a 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 over lines

for i in range ( len (X)):

# iterate over columns

for j in range ( len (X [ 0 ])):

result [i] [j] = X [i] [j] + Y [i] [j]

for r in result:

print (r)

Exit :

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

Explanation: —
In this program, we used nested for loops to iterate over every row and every column. At each point, we add the corresponding elements to two matrices and store them as a result.

Usage understanding nested

# Program for adding two matrices
# using comprehension list

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)

Output:

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

Explanation: — The output of this program is the same as above. We used a nested list comprehension to iterate over each element in the matrix.