PostgreSQL
 sql >> Database >  >> RDS >> PostgreSQL

Come aggregare coppie corrispondenti in componenti connessi in Python

Con networkX:

import networkx as nx
G1=nx.Graph()
G1.add_edges_from([("a","b"),("b","c"),("c","d"),("d","e"),("f","g")])
sorted(nx.connected_components(G1), key = len, reverse=True)

dando:

[['a', 'd', 'e', 'b', 'c'], ['f', 'g']]

Devi controllare l'algoritmo più veloce ora ...

OP:

Funziona alla grande! Ho questo nel mio database PostgreSQL ora. Organizza semplicemente le coppie in una tabella a due colonne, quindi usa array_agg() per passare alla funzione PL/Python get_connected() . Grazie.

CREATE OR REPLACE FUNCTION get_connected(
    lhs text[],
    rhs text[])
  RETURNS SETOF text[] AS
$BODY$
    pairs = zip(lhs, rhs)

    import networkx as nx
    G=nx.Graph()
    G.add_edges_from(pairs)
    return sorted(nx.connected_components(G), key = len, reverse=True)

$BODY$ LANGUAGE plpythonu;

(Nota:ho modificato la risposta, poiché pensavo che mostrare questo passaggio potesse essere un'aggiunta utile, ma troppo lunga per un commento.)