get and set diagonal of coo_matrix