w3resource
NumPy Exercises

NumPy: Create a 3x3 identity matrix, i.e. diagonal elements are 1,the rest are 0

NumPy: Basic Exercise-27 with Solution

Write a NumPy program to create a 3x3 identity matrix, i.e. diagonal elements are 1,the rest are 0.

Sample Solution :

Python Code :

import numpy as np
x = np.eye(3)
print(x)

Sample Output:

[[ 1.  0.  0.]
 [ 0.  1.  0.]
 [ 0.  0.  1.]]                         

Pictorial Presentation:

NumPy: Create a 3x3 identity matrix, i.e. diagonal elements are 1,the rest are 0.

Python Code Editor:


Have another way to solve this solution? Contribute your code (and comments) through Disqus.

Previous: Write a NumPy program to find the number of rows and columns of a given matrix.
Next: Write a NumPy program to create a 10x10 matrix, in which the elements on the borders will be equal to 1, and inside 0.

What is the difficulty level of this exercise?