[graph-tool] Building (and working with) bipartite graphs

Davide Cittaro cittaro.davide at gmail.com
Thu Jul 9 14:37:34 CEST 2020


Hi, 
I'm trying to build a bipartite graph from a MxN array. So far I've been able to do this:

# adjacency is the original np.array

S = []
E = []
W = []
for x in range(adjacency.shape[0]):
    for y in np.where(adjacency[x] > 0)[0]:
        S.append(x)
        E.append(y + adjacency.shape[0]) # so that node index starts with an offset)
        W.append(adjacency[x, y])

g = gt.Graph(directed=True)

g.add_edge_list(np.hstack([np.array(S)[:, None], np.array(E)[:, None]]))

ew = g.new_edge_property("double")
ew.a = W
g.ep['weight'] = ew

recs = [g.ep.weight]
rec_types = ['real-exponential']

state = gt.NestedBlockState(g, state_args=dict(recs=recs,
                                    rec_types=rec_types))

[…]

I'm pretty sure this is not the most efficient way to go, but gt.is_bipartite(g) returns True. Can anybody suggest a more straightforward approach? Also, is it necessary to specify a VertexPropertyMap with node type (i.e. M or N)?
Anyhow, after I had the graph and the final model, I tried to draw:

gt.draw_hierarchy(state, layout='bipartite')

but I get this error

IndexError
                                Traceback (most recent call last)

<ipython-input-67-d7655416181e> in <module>
----> 1 gt.draw_hierarchy(state, layout='bipartite')



~/anaconda3/envs/schist/lib/python3.8/site-packages/graph_tool/draw/cairo_draw.py in draw_hierarchy(state, pos, layout, beta, node_weight, vprops, eprops, hvprops, heprops, subsample_edges, rel_order, deg_size, vsize_scale, hsize_scale, hshortcuts, hide, bip_aspect, empty_branches, **kwargs)
   1773
                                   rel_order_leaf=True)

   1774     elif layout == "bipartite":
-> 1775
         tpos = get_bip_hierachy_pos(state, aspect=bip_aspect,

   1776
                                     node_weight=node_weight)

   1777         tpos = t.own_property(tpos)



~/anaconda3/envs/schist/lib/python3.8/site-packages/graph_tool/draw/cairo_draw.py in get_bip_hierachy_pos(state, aspect, node_weight)
   2145         t.add_edge(root, p2)
   2146         for p in ps:
-> 2147             if bc.a[tb[p]] == 0:
   2148                 t.add_edge(p2, p)
   2149             else:



IndexError
: index 1 is out of bounds for axis 0 with size 1



Any help is appreciated

d



More information about the graph-tool mailing list