import numpy as np
# create a 3x3 matrix filled with zeros
matrix = np.zeros((3, 3))
# fill the diagonal with ones
np.fill_diagonal(matrix, 1)
print(matrix)