我有一个数组X，其中node的值代表两个组.如果我有n组如何给每一组分配唯一的颜色?

Question

My array

X = [1, 0, 2, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1]

我正在画一个图，其中每个节点X [I]相互连接。我想给属于一个特定组的所有节点分配一个唯一的颜色。

你能建议一个Python代码来分配颜色吗?我试过这种方法，但不能推广到n组。

color_map = []
for i in X:
if i==0:
color_map.append('blue')
elif i==1:
color_map.append('yellow')
else: 
color_map.append('green')

Answer 1

创建一个颜色映射，然后将每个数字转换为对应的颜色:

colors = {
0: 'blue',
1: 'yellow',
2: 'green',
}
mapped = [colors[i] for i in X]

要扩展该解决方案，只需在颜色映射中添加更多的值。