Início / coding / Perguntas / 79553487
Accepted
alpacafondue
Asked: 2025-04-04 00:17:33 +0800 CST

Criar um gráfico de rede agregado a partir do dataframe do Pandas contendo uma coluna com uma lista de nós para cada linha?

  • 772

Estou dando uma olhada em visualizações de rede em Python. Tenho um dataframe como o seguinte:

| user | nodes    |
| -----| ---------|
| A    | [0, 1, 3]|
| B    | [1, 2, 4]|
| C    | [0, 3]   |
|...   |          |

Existe uma maneira de plotar facilmente um gráfico de rede (NetworkX?) a partir de dados que contenham a lista de nós em cada linha? A presença de um nó em uma linha aumentaria a proeminência desse nó no gráfico (ou a proeminência/peso da aresta no relacionamento entre dois nós).

insira a descrição da imagem aqui

Suponho que alguma transformação seria necessária para colocar os dados no formato apropriado para que o NetworkX (ou similar) pudesse criar os relacionamentos do gráfico.

Obrigado!

python

1 respostas

  1. Best Answer

    Como você tem listas, usar pandas não seria mais eficiente.

    Você pode usar itertoolspara enumerar as arestas e collections.Countercontá-las, depois construir o gráfico e plotar com uma largura baseada no peso:

    from itertools import combinations, chain
from collections import Counter
import networkx as nx

c = Counter(chain.from_iterable(combinations(sorted(l), 2) for l in df['nodes']))

G = nx.Graph()
G.add_weighted_edges_from((*e, w) for e, w in c.items())

pos = nx.spring_layout(G)
nx.draw_networkx(G, pos)

for *e, w in G.edges(data='weight'):
    nx.draw_networkx_edges(G, pos, edgelist=[e], width=w)

    Saída:

    gráfico networkx de arestas ponderadas

    Entrada usada:

    df = pd.DataFrame({'user': ['A', 'B', 'C'],
                   'nodes': [[0, 1, 3], [1, 2, 4], [0, 3]],
                  })
    • 0

relate perguntas