networkx - change node size based on list or dictionary value

Question

I'm trying to make a graph in networkx. I'm having trouble assigning different node sizes to the nodes. Here is my code I've been playing with: ``` import sys from collections import defaultdict import networkx as nx import matplotlib.pyplot as plt inp = sys.argv[1] cluster = sys.argv[1] + ".cluster" counts = sys.argv[1] + ".counts" with open(cluster, "r") as f1: edges = [line.strip().split('	') for line in f1] with open(counts, "r") as f2: countsdic = defaultdict(list) for line in f2: k,v = line.strip().split() countsdic[k].append(v) tmp = [] for el in sum(edges, []): tmp.append(el) nodes = [] for t in tmp: if t not in nodes: nodes.append(t) node_sizes = {} for n in nodes: node_sizes[n] = ' '.join(countsdic[n]) print node_sizes nodes2 = [] sizes = [] for k in node_sizes.keys(): nodes2.append(k) for v in node_sizes.values(): sizes.append(v) print nodes2 print len(nodes2) print sizes print len(sizes) g = nx.Graph() g.add_nodes_from(nodes) g.add_edges_from(edges) nx.draw_random(g, node…

Accepted Answer

2014/07/08 12:29PM: Updated to reflect comments from @user3358205

The problem is that the drawing functions in NetworkX require `node_sizes` to be input as a `list` of `int`s, while you are passing a `list` of strings. You can read the parameters to the drawing functions here.

Because I don't have the input files to your program, I can't reproduce your output. However, here's an example where you vary the size of the nodes by passing a `list` of `node_sizes`. Note that in the output, I am labeling each node by their size.

```
import sys, networkx as nx, matplotlib.pyplot as plt

# Create a list of 10 nodes numbered [0, 9]
nodes = range(10)
node_sizes = []
labels = {}
for n in nodes:
        node_sizes.append( 100 * n )
        labels[n] = 100 * n

# Node sizes: [0, 100, 200, 300, 400, 500, 600, 700, 800, 900]

# Connect each node to its successor
edges = [ (i, i+1) for i in range(len(nodes)-1) ]

# Create the graph and draw it with the node labels
g = nx.Graph()
g.add_nodes_from(nodes)
g.add_edges_from(edges)

nx.draw_random(g, node_size = node_sizes, labels=labels, with_labels=True)    
plt.show()
```

networkx - change node size based on list or dictionary value

Solution

Problem