NumPy - linalg.solve() function

The NumPy linalg.solve() function is used to solve a linear matrix equation, or system of linear scalar equations. The syntax for using this function is given below:

Syntax

numpy.linalg.solve(a, b)

Parameters

Return Value

Returns solution to the system a x = b. Returned shape is identical to b.

Exception

Raises LinAlgError exception, if a is singular or not square.

Example: Solving system of linear equations

Consider the below mentioned system of linear equations:

Which can be represented as below in the matrix form.

The solution of the above equations is: x=1, y=5, z=7. This can be solved using numpy.linalg.solve() function as shown in the example below.

import numpy as np
A = np.array([[1, 2, 3], 
              [2, 3, 1],
              [3, 1, 2]])
B = np.array([32, 24, 22])

sol = np.linalg.solve(A, B)
print(sol)

The output of the above code will be:

[ 1.  5.  7.]