sql >> Base de Datos >  >> RDS >> PostgreSQL

Cómo agregar pares coincidentes en componentes conectados en Python

Con redX:

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']]

Tienes que comprobar el algoritmo más rápido ahora...

OP:

¡Esto funciona muy bien! Tengo esto en mi base de datos PostgreSQL ahora. Simplemente organice los pares en una tabla de dos columnas, luego use array_agg() para pasar a la función PL/Python get_connected() . Gracias.

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:edité la respuesta, ya que pensé que mostrar este paso podría ser un apéndice útil, pero demasiado largo para un comentario).