如何从一个邻接矩阵为特定节点创建网络图?

2024-05-15 17:54:49 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个邻接矩阵5000X5000,我想创建一个网络图。需求是用户将输入节点,输出将是该特定输入节点的图(一阶和二阶)。在

我已经尝试过使用Gephi,但是由于邻接矩阵很大,我不能集中精力在每个节点上。所以我想如果我可以为特定的节点创建一个图(因为我只对每个节点的一阶和二阶连接感兴趣,而不超过这个值)

Gephi是基于用户界面的,所以我没有代码。在

输入将是一个节点标识,输出将是一个对应于该节点标识的图形(一级和二级连接)


Tags: 代码用户图形节点用户界面标识感兴趣网络图
1条回答
网友
1楼 · 发布于 2024-05-15 17:54:49

下面是一个使用networkx的实现:

import networkx as nx
import numpy as np

# make dummy adjacency matrix
a = np.random.rand(100,100)
a = np.tril(a)
a = a>0.95

# make graph from adjaceny matrix
G = nx.from_numpy_matrix(a)


def neigh(G, node, depth):
    """ given starting node, recursively find neighbours
        until desired depth is reached
    """

    node_list = []
    if depth==0:
        node_list.append(node)
    else:
        for neighbor in G.neighbors(node):
            node_list.append(node)
            node_list += neigh(G, neighbor, depth-1)
    return list(set(node_list)) # intermediate conversion to set to lose duplicates. 

# a bit more compressed:
def neigh_short(G, node, depth):
    """ given starting node, recursively find neighbours
        until desired depth is reached
    """

    node_list = [node]
    if depth>0:
        for neighbor in G.neighbors(node)
            node_list += neigh_short(G, neighbor, depth-1)
    return list(set(node_list)) # intermediate conversion to set to lose duplicates. 

# example:
# find all neighbours with distance 2 from node 5:
n = neigh(G, node=5, depth=2)

# extract the respective subgraph from G and store in H
H = G.subgraph(n)

相关问题 更多 >