(nx_graph, vmin=0, vmax=7)
| 77 | |
| 78 | |
| 79 | def graph_colors(nx_graph, vmin=0, vmax=7): |
| 80 | cnorm = mcol.Normalize(vmin=vmin, vmax=vmax) |
| 81 | cpick = cm.ScalarMappable(norm=cnorm, cmap="viridis") |
| 82 | cpick.set_array([]) |
| 83 | val_map = {} |
| 84 | for k, v in nx.get_node_attributes(nx_graph, "weight").items(): |
| 85 | val_map[k] = cpick.to_rgba(v) |
| 86 | colors = [] |
| 87 | for node in nx_graph.nodes(): |
| 88 | colors.append(val_map[node]) |
| 89 | return colors |
| 90 | |
| 91 | |
| 92 | def draw_graph( |
no outgoing calls
no test coverage detected