networkx节点node样式style定制化，Python

清疚 2023-10-01 08:08 9阅读 0赞

import networkx as nx
    from matplotlib import pyplot as plt
    
    KEY = 'key'
    
    
    def app():
        G = nx.DiGraph()
        plt.figure(figsize=(12, 10), dpi=100)
    
        G.add_node(G.number_of_nodes())
        G.add_node(G.number_of_nodes())
        G.add_node(G.number_of_nodes())
    
        G.nodes[0][KEY] = [(90, 1), (80, 2), (70, 3)]
        G.nodes[1][KEY] = [(91, 1), (81, 2), (71, 3)]
        G.nodes[2][KEY] = [(92, 1), (82, 2), (72, 3)]
    
        print(G.nodes(data=True))
    
        pos = nx.spring_layout(G)
    
        labels = {}
        for i in range(G.number_of_nodes()):
            ls = G.nodes[i][KEY]
            s = str(i) + '\n' + tuplelist2str(ls)
            labels.__setitem__(i, s)
    
        nx.draw_networkx_nodes(G, pos=pos,
                               node_color='red',
                               node_size=2000,
                               node_shape='s')
    
        label_options = {"ec": "black", "fc": "white", "alpha": 0.7}
        nx.draw_networkx_labels(G, pos=pos,
                                labels=labels,
                                font_size=8,
                                font_color='blue',
                                bbox=label_options)
    
        plt.margins(0.1)
        plt.axis('off')
        plt.show()
    
    
    def tuplelist2str(tl):
        s = ''
        for t in tl:
            s = s + '(' + str(t[0]) + ',' + str(t[1]) + ')'
        return s
    
    
    if __name__ == '__main__':
        app()

输出：

![watermark_type_d3F5LXplbmhlaQ_shadow_50_text_Q1NETiBAemhhbmdwaGls_size_20_color_FFFFFF_t_70_g_se_x_16][]

[watermark_type_d3F5LXplbmhlaQ_shadow_50_text_Q1NETiBAemhhbmdwaGls_size_20_color_FFFFFF_t_70_g_se_x_16]: https://img-blog.csdnimg.cn/cf731ca98af1415981cb1e332871920d.png?x-oss-process=image/watermark,type_d3F5LXplbmhlaQ,shadow_50,text_Q1NETiBAemhhbmdwaGls,size_20,color_FFFFFF,t_70,g_se,x_16