Skip to content

Commit

Permalink
add a method
Browse files Browse the repository at this point in the history
  • Loading branch information
bdpedigo committed Mar 20, 2024
1 parent 2ed2d09 commit fe4785d
Showing 1 changed file with 16 additions and 0 deletions.
16 changes: 16 additions & 0 deletions networkframe/networkframe.py
Original file line number Diff line number Diff line change
Expand Up @@ -1244,6 +1244,22 @@ def condense(
nodes = pd.DataFrame(index=nodes_index)
return self.__class__(nodes, edges, directed=self.directed)

def sort_spectral(self, weight_col="weight", inplace=False) -> "NetworkFrame":
adjacency = self.to_sparse_adjacency(weight_col=weight_col)
adjacency = adjacency + adjacency.T
adjacency = adjacency.astype(float)

# from scipy.sparse.linalg import svds
# u, s, vh = svds(adjacency, k=1)

from scipy.sparse.linalg import eigsh

_, u = eigsh(adjacency, k=1, which="LM", return_eigenvectors=True)

nodes = self.nodes.iloc[np.argsort(u[:, 0])]

return self._return(nodes=nodes, inplace=inplace)


class LocIndexer:
"""A class for indexing a NetworkFrame using .loc."""
Expand Down

0 comments on commit fe4785d

Please sign in to comment.