如何用NetworkX为节点设置多颜色？以空手道俱乐部图为例

嘿，完全可以实现！NetworkX搭配Matplotlib就能轻松搞定你说的空手道俱乐部图的边着色和单个节点多颜色需求，我给你拆解具体的实现方法：

边着色是NetworkX里比较容易实现的需求，你可以根据边的属性（比如连接的节点所属俱乐部、边的权重等）给不同边分配不同颜色。以空手道俱乐部图为例，我们可以给同俱乐部节点间的边设为蓝色，跨俱乐部的边设为红色：

import networkx as nx
import matplotlib.pyplot as plt

# 加载经典的空手道俱乐部图
G = nx.karate_club_graph()

# 遍历所有边，根据节点所属俱乐部分配颜色
edge_colors = []
for u, v in G.edges():
    if G.nodes[u]['club'] == G.nodes[v]['club']:
        edge_colors.append('blue')  # 同俱乐部边用蓝色
    else:
        edge_colors.append('red')   # 跨俱乐部边用红色

# 设定图的布局（这里用弹簧布局，seed固定布局方便重复查看）
pos = nx.spring_layout(G, seed=42)

# 依次绘制边、节点、标签
nx.draw_networkx_edges(G, pos, edge_color=edge_colors, width=2)
nx.draw_networkx_nodes(G, pos, node_color='lightgray', node_size=600)
nx.draw_networkx_labels(G, pos, font_size=10)

plt.axis('off')
plt.show()

NetworkX默认的节点绘制是单一颜色，但我们可以借助Matplotlib的底层绘图功能，通过叠加图形或者自定义节点标记来实现单个节点多颜色的效果，这里给你两种常用方法：

方法1：叠加扇形实现分色节点 #

如果只是需要节点分成几个颜色区域（比如左右/上下分色），可以用Matplotlib的Wedge（扇形）组件叠加绘制：

import networkx as nx
import matplotlib.pyplot as plt
from matplotlib.patches import Wedge

G = nx.karate_club_graph()
pos = nx.spring_layout(G, seed=42)

# 先绘制基础的灰色节点和标签
nx.draw_networkx_nodes(G, pos, node_color='lightgray', node_size=1200)
nx.draw_networkx_labels(G, pos, font_size=10)

# 给所有节点做左右分色（左红右蓝）
for node in G.nodes():
    x, y = pos[node]
    # 扇形半径0.05适配布局角度范围
    left_wedge = Wedge((x, y), 0.05, 90, 270, color='red')
    right_wedge = Wedge((x, y), 0.05, 270, 90, color='blue')
    plt.gca().add_patch(left_wedge)
    plt.gca().add_patch(right_wedge)

# 绘制边
nx.draw_networkx_edges(G, pos, edge_color='gray', width=1)

plt.axis('off')
plt.show()

方法2：自定义图像作为多颜色节点 #

如果需要更复杂的多颜色节点（比如渐变、多色块组合），可以用PIL生成自定义图像，再作为节点标记绘制：

import networkx as nx
import matplotlib.pyplot as plt
from PIL import Image
import numpy as np

G = nx.karate_club_graph()
pos = nx.spring_layout(G, seed=42)

# 生成一个30x30的多颜色图像（上红下蓝，透明背景）
custom_node_img = Image.new('RGBA', (30, 30), (255, 255, 255, 0))
for x in range(30):
    for y in range(30):
        if y < 15:
            custom_node_img.putpixel((x, y), (255, 0, 0, 255))  # 上半部分红色
        else:
            custom_node_img.putpixel((x, y), (0, 0, 255, 255))  # 下半部分蓝色

# 转换为numpy数组，方便Matplotlib调用
img_array = np.array(custom_node_img)

# 先绘制边和标签
nx.draw_networkx_edges(G, pos, edge_color='gray', width=1)
nx.draw_networkx_labels(G, pos, font_size=10)

# 逐个绘制自定义多颜色节点
for node in G.nodes():
    x, y = pos[node]
    # 调整图像范围适配节点位置，zorder确保节点在最上层
    plt.imshow(img_array, extent=(x-0.06, x+0.06, y-0.06, y+0.06), zorder=10)

plt.axis('off')
plt.show()

这些方法都能灵活满足你的需求，边着色可以直接通过NetworkX的API实现，节点多颜色则需要结合Matplotlib的绘图能力来扩展~

内容的提问来源于stack exchange，提问作者user137927