代码之家 › 专栏 › 技术社区 › mathfux

有没有一种简单的方法可以在有向图中创建层次标签名?

networkx python

mathfux · 技术社区 · 4 年前

我想创建networkx节点的层次映射。有向图。如右图所示:

import networkx as nx
import matplotlib.pyplot as plt
from networkx.drawing.nx_agraph import graphviz_layout
H = nx.DiGraph([(0, 1), (0, 12), (1, 2), (1, 3), (3, 4), (3, 7), (4, 5), (4, 6), 
                (7, 8), (7, 9), (9, 10), (9, 11), (12, 13), (12, 14)])
kwargs = {'pos': graphviz_layout(H, prog='dot'), 'nodelist':[], 
          'with_labels':True, 'bbox': dict(boxstyle='round,pad=0.7')}

fig = plt.figure(figsize=(20,10))
fig.add_subplot(1, 2, 1)
nx.draw(H, **kwargs)
fig.add_subplot(1, 2, 2)
nx.draw(H, **kwargs, labels=labels)
plt.show()

价值 labels 我在脚本中使用的是:

{0: (),
 1: (0,),
 2: (0, 0),
 3: (0, 1),
 4: (0, 1, 0),
 5: (0, 1, 0, 0),
 6: (0, 1, 0, 1),
 7: (0, 1, 1),
 8: (0, 1, 1, 0),
 9: (0, 1, 1, 1),
 10: (0, 1, 1, 1, 0),
 11: (0, 1, 1, 1, 1),
 12: (1,),
 13: (1, 0),
 14: (1, 1)}

有没有一个简单的方法来创建这样的标签 networkx 或者其他包裹?

0 回复 | 直到 4 年前

Paul Brodersen 4 年前

这应该适用于任意树,但可以简化为二叉树,如示例所示。

import networkx as nx

H = nx.DiGraph([(0, 1), (0, 12), (1, 2), (1, 3), (3, 4), (3, 7), (4, 5), (4, 6),
                (7, 8), (7, 9), (9, 10), (9, 11), (12, 13), (12, 14)])

# iterate over nodes starting at the root
order = nx.topological_sort(H)
root = next(order)
labels = {root : list()}

for node in order:
    parent_node = next(H.predecessors(node))
    parent_label = labels[parent_node]

    # check for labeled siblings before creating a new label
    ii = 0
    while parent_label + [ii] in labels.values():
        ii += 1
    labels[node] = parent_label + [ii]

print(labels)
# {0: [], 12: [0], 14: [0, 0], 13: [0, 1], 1: [1], 3: [1, 0], 7: [1, 0, 0], 9: [1, 0, 0, 0], 11: [1, 0, 0, 0, 0], 10: [1, 0, 0, 0, 1], 8: [1, 0, 0, 1], 4: [1, 0, 1], 6: [1, 0, 1, 0], 5: [1, 0, 1, 1], 2: [1, 1]}

推荐文章

L. Hiao · 如何从csv邻接矩阵数据帧创建有向图?

8 年前

ccc · 访问列表Python中的元素

8 年前

ccc · 顶点收缩-python

8 年前

Toke Faurby · 如何在networkx中获取色条。draw\u networkx?

8 年前

ccc · 根据节点之间的欧几里德距离指定边权重

8 年前

nick88 · 在NetworkX中同时创建属性和边(如果存在相同的属性)

8 年前

Fazeleh · 如何通过networkx在图形上设置垂直标签

8 年前

Arash Ashrafzadeh · 如何在Python中生成不同类型的图?

8 年前

NicolaiF · 获取networkx draw\u网络对象的x、y位置

8 年前

eric lardon · 在随机生成的图中获取节点位置

8 年前