NumPy matrix operations

A matrix is a two-dimensional array of numbers, symbols, or expressions arranged in rows and columns. Matrices play a crucial role in representing and solving various mathematical problems, from linear transformations to solving systems of equations.

Introduction to NumPy

NumPy is a fundamental Python library for numerical computations, supporting working with large, multi-dimensional arrays and matrices. It offers various mathematical functions to perform various operations on these arrays efficiently.

Learn about matrix operations.

Creating matrices with NumPy

We can create matrices using NumPy's array creation functions. For example:

import numpy as np
# Creating a 3x3 matrix
matrix = np.array([[1, 2, 3],
                   [4, 5, 6],
                   [7, 8, 9]])
                   
print("Matrix A:")
print(matrix)

Matrix operations

Addition and subtraction

We can perform matrix addition and subtraction element-wise.

import numpy as np
A = np.array([[1, 2],
              [3, 4]])
B = np.array([[5, 6],
              [7, 8]])
addition = A + B
subtraction = A - B
print("Addition of matrices:")
print(addition)
print("Subtraction of matrices:")
print(subtraction)

Scalar multiplication

We can multiply a matrix by a scalar value.

import numpy as np
A = np.array([[1, 2],
              [3, 4]])
scalar = 2
scalar_multiplication = scalar * A
print("Scalar multiplication:")
print(scalar_multiplication)

Transposition

Transposing a matrix flips it over its diagonal.

import numpy as np
A = np.array([[1, 2],
              [3, 4]])
transpose = np.transpose(A)
print("Original matrix A:")
print(A)
print("Transpose of matrix A :")
print(transpose)

Matrix multiplication

The dot product is different from element-wise multiplication. Dot product yields a scalar or matrix, whereas element-wise multiplication results in a matrix with products of corresponding elements.

import numpy as np
A = np.array([[1, 2],
              [3, 4]])
              
B = np.array([[5, 6],
              [7, 8]])
              
dot_product = np.dot(A, B)
element_wise = A * B
print("Dot product:")
print(dot_product)
print("Element-wise multiplication:")
print(element_wise)

Determinant of a matrix

We can easily determine the determinant of a square matrix by utilizing the np.linalg.det() function, which calculates the determinant of the provided matrix.

import numpy as np
A = np.array([[1, 2],
              [3, 4]])
determinant = np.linalg.det(A)
print("Determinant of matrix A:")
print(determinant)

Rank of a matrix

We can determine the rank of a matrix by using the np.linalg.matrix_rank() function, which calculates the rank of the provided matrix.

import numpy as np
A = np.array([[1, 2, 3],
              [4, 5, 6],
              [7, 8, 9]])
rank = np.linalg.matrix_rank(A)
print("Rank of matrix A:")
print(rank)

Conclusion

NumPy provides a wide range of matrix operations, from simple arithmetic to advanced linear algebra and data analysis tasks. With a solid grasp of NumPy's capabilities, we can confidently navigate matrix arithmetic, advanced linear algebra, and data analysis.

Free Resources

Copyright ©2026 Educative, Inc. All rights reserved