лучший способ сопоставить числа с цветами из столбца массива numpy

Question

в zs [:, 2] - это числа в моем случае, например, от 1 до 5.

plt.scatter(zs[:,0], zs[:,1], c=zs[:,2], alpha=0.5)

Как лучше всего отобразить эти числа (из zs [:, 2]) в Speci c цвета?

Answer 1

Я сделал это аналогичным образом с помощью panda lib ( Как мне преобразовать числа в цветовую шкалу в matplotlib? ).

def col_map(x):
    print(x)
    if x == 0.0:
        return "green"
    if x == 1.0:
        return 'red'
    else:
        return 'blue'

df = pd.DataFrame({"x":zs[:,0],"y":zs[:,1],"c":[col_map(x) for x in zs[:,2].tolist()]})
plt.scatter(df.x, df.y, color=df.c)