How to get Minimum Spanning Tree Matrix in python

Question

Initially, i have 2d array. By using this array i have created a graph with weight on its edges. Now i am trying to use this Graph to make Minimum Spanning Tree matrix but i cant make it as desire. I am using the following code to make graph.

 G = nx.from_numpy_matrix(ED_Matrix, create_using=nx.DiGraph)
 layout = nx.spring_layout(G)
 sizes = len(ED_Matrix)
 nx.draw(G, layout, with_labels=True, node_size=sizes)
 labels = nx.get_edge_attributes(G, "weight")
 output = nx.draw_networkx_edge_labels(G, pos=layout, edge_labels=labels)
 plt.show()

And its gives the output like this

Now i am using MST code, to get the its MST matrix but its gives error like this.

 from scipy.sparse import csr_matrix
 from scipy.sparse.csgraph import minimum_spanning_tree
 Tcsr = minimum_spanning_tree(G)
 Tcsr.toarray().astype(int)

mathfux · Accepted Answer · 2020-09-10 12:59:15Z

1

Taking into account example from docs of scipy, it should be constructed from adjacency matrix of G, not from G.

You might want to replace G with nx.adjacency_matrix(G) or csr_matrix(nx.adjacency_matrix(G)) or ED_Matrix itself in calculation (assignment) of Tcsr:

Tcsr = minimum_spanning_tree(nx.adjacency_matrix(G)) #or
Tcsr = minimum_spanning_tree(csr_matrix(nx.adjacency_matrix(G))) #or
Tcsr = minimum_spanning_tree(ED_Matrix)

Tcsr is a sparse matrix which is later converted to numpy array.

edited Sep 10, 2020 at 12:59

answered Sep 9, 2020 at 21:23

mathfux

5,9792 gold badges21 silver badges38 bronze badges

Sign up to request clarification or add additional context in comments.

3 Comments

robinHood013 Over a year ago

ED_Matrix , is just a 2d array, its not a calculation of Tcsr

mathfux Over a year ago

I think I was not clear mentioning calculation. I mean it's an assignment.

robinHood013 Over a year ago

I have replaced the G with nx.adjacency_matrix(G) plus instead of using Tcsr.toarray().astype(int) i have used Tcsr.toarray().astype(float). Now its working

Collectives™ on Stack Overflow

How to get Minimum Spanning Tree Matrix in python

1 Answer 1

3 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

3 Comments

Your Answer

Sign up or log in

Post as a guest

Related