Python `matplotlib` 图形的彩色输出

发布于06月08日

如何使用python的matplotlib库输出彩色图形(每个顶点都有自己的 colored颜色 )？是否有任何方法可以调整每个顶点的特定 colored颜色？

不使用 colored颜色的代码示例:

import networkx as nx
import matplotlib.pyplot as plt

class Graph:
    def __init__(self, edges):
        self._edges = edges

    def visualize(self):
        vg = nx.Graph()
        vg.add_edges_from(self._edges)
        nx.draw_networkx(vg)
        plt.show()

nodes = [['A', 'B'], ['A', 'C'], ['B', 'D'], ['C', 'D'],
         ['C', 'E'], ['D', 'F'], ['E', 'F']]
G = Graph(nodes)
G.visualize()

这就是我想看到的:

推荐答案

我不确定您是只想为这种情况更改 colored颜色，还是想让它更灵活—使用list comprehension，但AFAIK draw_networkx有一个参数，它接受字符串列表或RGB个浮点元组，所以您只能准备一个 colored颜色列表:

import networkx as nx
import matplotlib.pyplot as plt

class Graph:
    def __init__(self, edges, colors):
        self._edges = edges
        self._colors = colors

    def visualize(self):
        vg = nx.Graph()
        vg.add_edges_from(self._edges)    
        nx.draw_networkx(vg, node_color=self._colors)
        plt.show()



nodes = [['A', 'B'], ['A', 'C'], ['B', 'D'], ['C', 'D'],
         ['C', 'E'], ['D', 'F'], ['E', 'F']]

colors = ['green', 'red', 'red', 'green', 'green', 'red']

G = Graph(nodes, colors)
G.visualize()