How to increase node spacing for networkx.spring_layout
Asked Answered
P

3

55

Drawing a clique graph with

import networkx as nx
....
nx.draw(G, layout=nx.spring_layout(G))

produces the following picture:

enter image description here

Obviously, the spacing between the nodes (e.g., the edge length) needs to be increased. I've googled this and found this suggestion here:

For some of the layout algorithms there is a scale parameter that might help. e.g.

import networkx as nx
G = nx.path_graph(4)
pos = nx.spring_layout(G)  # default to scale=1
nx.draw(G, pos)
pos = nx.spring_layout(G, scale=2)  # double distance between all nodes
nx.draw(G, pos)

However, the scale parameter does not seem to have any effect.

What is the right method to get a better drawing?

Padrone answered 11/1, 2013 at 17:19 Comment(1)
Apart from the answer provided below, I have also found that increasing the figure size helps.Kissinger
G
73

The solution to this problem lies in the NetworkX version 1.8 which is yet to be released, but is available via git hub.

Do the following to increase the distance between nodes:

pos = nx.spring_layout(G, k=0.15, iterations=20)
# k controls the distance between the nodes and varies between 0 and 1
# iterations is the number of times simulated annealing is run
# default k=0.1 and iterations=50

Tweak with these parameters to see how it works. But despite this there is no guarantee that all nodes are non-overlapping

Gratulate answered 25/6, 2013 at 6:16 Comment(2)
According to networkx.org/documentation/stable/reference/generated/…, k=1/n^(1/2), so it's not stable at 0.1Sheppard
This answer didn't help in my case, which is similar, however, OrangeSherbet's answer did work.Keynesianism
W
20

The real answer to your question is that your original graph is not a single, fully-connected component. It is three separate graphs.

What is happening is the three pieces are flying away to infinity, which upon re-scaling makes each component look like a tiny blob.

The algorithm spring_layout specifies a repulsive force between all nodes (anti-gravity), and an attractive force between only connected nodes (the "springs").

Thus, if the graph isn't connected, the individual pieces will fly away from the repulsive force since there is nothing connecting them. There are two options: Change the force law (edit the networkx code), or graph the components separately.

Here's how to add a force that attracts all nodes to the center of the chart. Add the last line of this code snippet to def _fruchterman_reingold in layouts.py:

# displacement "force"
displacement = np.einsum('ijk,ij->ik',
                         delta,
                         (k * k / distance**2 - A * distance / k))
# ADD THIS LINE - prevent things from flying off into infinity if not connected
displacement = displacement - pos / ( k * np.sqrt(nnodes))

This single line of code allows you to make charts like: enter image description here rather than enter image description here However the "best" way to deal with this is to chart the components seperately.

You can iterate over the components, charting them in seperate plots, with the functions described here.

See this github issue for more discussion.

Weinstock answered 26/2, 2019 at 0:57 Comment(0)
O
17

I used the optimal distance parameter of the Kamada Kawai layout, and set the distance between non-connected components to the maximum distance in the graph. There is probably a better way of munging the dictionaries, but this is pretty easy:

df = pd.DataFrame(index=G.nodes(), columns=G.nodes())
for row, data in nx.shortest_path_length(G):
    for col, dist in data.items():
        df.loc[row,col] = dist

df = df.fillna(df.max().max())

layout = nx.kamada_kawai_layout(G, dist=df.to_dict())
Ogee answered 26/4, 2018 at 16:36 Comment(0)

© 2022 - 2024 — McMap. All rights reserved.