import numpy as np # create a 3x3 identity matrix matrix = np.identity(3) print(matrix)