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

Question

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

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

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

Answer 1

我通过 panda lib (How can I convert numbers to a color scale in 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)