从numpy数组列将数字映射到颜色的最佳方法

在zs[：，2]中是我的例子中的数字，例如从1到5

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

---

将这些数字（从zs[：，2]）映射到特定颜色的最佳方法是什么？

我通过panda lib（）以类似的方式进行了映射

---

这回答了你的问题吗？将

cmap=plt.cm.jet

或另一个cmap添加到散点图线。@BenT我不知道cmap，是时候阅读cmap了，谢谢！

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)