Python – Element wise Matrix Difference

When working with matrices represented as lists of lists in Python, you often need to compute the element-wise difference between two matrices. This operation subtracts corresponding elements from each matrix position.

Using Nested Loops with zip()

The most straightforward approach uses nested loops with Python's zip() function to iterate through corresponding elements ?

matrix_1 = [[3, 4, 4], [4, 3, 1], [4, 8, 3]]
matrix_2 = [[5, 4, 7], [9, 7, 5], [4, 8, 4]]

print("First matrix:")
print(matrix_1)
print("Second matrix:")
print(matrix_2)

result = []

for row_1, row_2 in zip(matrix_1, matrix_2):
    temp_row = []
    for element_1, element_2 in zip(row_1, row_2):
        temp_row.append(element_2 - element_1)
    result.append(temp_row)

print("Element-wise difference (matrix_2 - matrix_1):")
print(result)

First matrix:
[[3, 4, 4], [4, 3, 1], [4, 8, 3]]
Second matrix:
[[5, 4, 7], [9, 7, 5], [4, 8, 4]]
Element-wise difference (matrix_2 - matrix_1):
[[2, 0, 3], [5, 4, 4], [0, 0, 1]]

Using List Comprehension

A more concise approach uses list comprehension for cleaner, more Pythonic code ?

matrix_1 = [[3, 4, 4], [4, 3, 1], [4, 8, 3]]
matrix_2 = [[5, 4, 7], [9, 7, 5], [4, 8, 4]]

result = [[b - a for a, b in zip(row_1, row_2)] 
          for row_1, row_2 in zip(matrix_1, matrix_2)]

print("Element-wise difference using list comprehension:")
print(result)

Element-wise difference using list comprehension:
[[2, 0, 3], [5, 4, 4], [0, 0, 1]]

Using NumPy Arrays

For numerical computations, NumPy provides efficient matrix operations ?

import numpy as np

matrix_1 = [[3, 4, 4], [4, 3, 1], [4, 8, 3]]
matrix_2 = [[5, 4, 7], [9, 7, 5], [4, 8, 4]]

# Convert to NumPy arrays
arr_1 = np.array(matrix_1)
arr_2 = np.array(matrix_2)

# Element-wise subtraction
result = arr_2 - arr_1

print("NumPy element-wise difference:")
print(result)
print("Result type:", type(result))

NumPy element-wise difference:
[[2 0 3]
 [5 4 4]
 [0 0 1]]
Result type: <class 'numpy.ndarray'>

Comparison

Conclusion

Use nested loops for clarity when learning matrix operations. For production code, prefer NumPy arrays for better performance, or list comprehension for pure Python solutions.