In [3]:
def hierarchy_pos(G, root=None, width=1., vert_gap=0.4, vert_loc=0, xcenter=0.5):
pos = {}
def _hierarchy_pos(G, root, left, right, vert_loc, xcenter, pos):
pos[root] = (xcenter, vert_loc)
children = list(G.successors(root))
if len(children) != 0:
dx = (right - left) / len(children)
nextx = left + dx / 2
for child in children:
pos = _hierarchy_pos(G, child, nextx - dx/2, nextx + dx/2, vert_loc - vert_gap, nextx, pos)
nextx += dx
return pos
if root is None:
root = [n for n, d in G.in_degree() if d == 0][0]
return _hierarchy_pos(G, root, 0, width, vert_loc, xcenter, pos)
# Map titles to colors
name_to_title = pd.Series(df.Title.values, index=df.all_name).to_dict()
title_color_map = {
'CEO': 'gold',
'President': 'orange',
'Sr. MD': 'lightblue',
'MD': 'lightgreen',
'Dir.': 'skyblue',
'VP': 'violet',
'AS': 'lightpink',
'AN': 'lightgray'
}
node_colors = []
for node in G.nodes():
title = name_to_title.get(node, "")
for key in title_color_map:
if key in title:
node_colors.append(title_color_map[key])
break
else:
node_colors.append('white')
# Plot the graph with legend
plt.figure(figsize=(18, 14))
pos = hierarchy_pos(G, vert_gap=0.4)
nx.draw(
G, pos,
with_labels=False,
arrows=True,
node_size=3500,
node_color=node_colors,
edgecolors='black'
)
nx.draw_networkx_labels(
G, pos,
font_size=15,
bbox=dict(facecolor='white', edgecolor='none', pad=1.0)
)
# Add legend
legend_patches = [mpatches.Patch(color=color, label=title) for title, color in title_color_map.items()]
plt.legend(
handles=legend_patches,
title='Titles',
loc='lower center',
bbox_to_anchor=(0.5, -0.1),
ncol=4,
fontsize=14,
title_fontsize=18,
frameon=True
)
plt.title('Backdoor Strategies, Inc.', size=38)
plt.tight_layout()
plt.show()
<ipython-input-3-7d254f2a78fe>:72: UserWarning: This figure includes Axes that are not compatible with tight_layout, so results might be incorrect. plt.tight_layout()
