結果 : get diagonal of sparse matrix python