import networkx as nx def una_linia(xarxa, linia): g = nx.DiGraph() for i, f, l in xarxa.edges.data('linia'): if l[:2] == linia: g.add_edge(i, f) return g