PostgreSQL
 sql >> Base de Dados >  >> RDS >> PostgreSQL

Como agregar pares correspondentes em componentes conectados em Python


Com redeX:
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']]

Você tem que verificar o algoritmo mais rápido agora ...

OP:

Isso funciona muito bem! Eu tenho isso no meu banco de dados PostgreSQL agora. Apenas organize os pares em uma tabela de duas colunas e use array_agg() para passar para a função PL/Python get_connected() . Obrigado.
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:eu editei a resposta, pois pensei que mostrar esta etapa poderia ser um adendo útil, mas muito longo para um comentário.)