Modify matrix diagonal -- dolfinx version for A.ident_zeros()

You can find a version of ident_zeros() for DOLFINx in this post.