nx.draw_networkx( B, pos = nx.drawing.layout.bipartite_layout(B, B_first_partition_nodes), width = edge_widths*5) # Or whatever other display options you like