scipy.sparse.dok_matrix.count_nonzero
scipy.sparse.dok_matrix.dot
Returns the main diagonal of the matrix