颜色编码matplotlib标记

Question

我正在尝试对我的情节标记进行颜色编码.

该想法是存在x,y和z值 - 包含坐标的x和y,z包含对应于0和150之间的每个坐标(即,x,y对)的数字.

现在,我需要为每个坐标绘制一个正方形,我完成了pyplot.plot(x,y, ls='s'),我可以分别设置标记边缘和面部颜色.通常,所有这些方块都具有相同的颜色.

我想要的是根据z值改变颜色,例如

z = 0-30:黑色,30-60:蓝色,60-90:绿色,90-120:黄色,120-150:红色

我尝试将数据分箱到那些箱子,但我不知道如何将其映射到颜色并相应地绘制它.

有什么建议？谢谢!

Answer 1

将这些 X、Y 坐标存储在单独的 X、Y 中（每个 Z 范围一对），然后通过多条绘图线将它们全部绘制出来。

概念示例：

X1,Y1 = """store your Z0-30 here"""
X2,Y2 = """store your z30-60 here"""
X3,Y3 = """store your z60-z90 here"""
X4,Y4 = """store your z90-120 here"""
X5,Y5 = """store your z120-150 here"""
fig=plt.figure()
pyplot.plot(X1,Y1,color='black')
pyplot.plot(X2,Y2,color='blue')
pyplot.plot(X3,Y3,color='green')
pyplot.plot(X4,Y4,color='yellow')
pyplot.plot(X5,Y5,color='red')
plt.show()

我建议使用散点图，它听起来更适合您所描述的内容。在这种情况下，将 pyplot.plot 更改为 pyplot.scatter 。您可以使用 cmap 按 Z 轴为它们着色。您还可以将自定义标记分配给每个组，如此例中所示。