Matplotlib,如何更改绘图中头部和线条的颜色?

Matplotlib, how to change the colors for heads and lines on a plot?

我有 10 个设备通过电线连接,我用 matplotlib.plot 为这个系统制作了一个图形,其中这些设备的状态为 0。在 运行 代码之后,一个 new_status 4个级别出来。然后我试图根据新状态绘制系统。我的问题是我想不出一种方法来改变头部和线条的颜色。例如,设备 1 的新状态现在是 3,所以设备 1 的头部颜色和连接到设备 6 的线的颜色应该是红色,而设备 6 应该仍然是蓝色。我尝试了 if 循环,但我无法理解。

x=(1,2,3,4,5,1,2,3,4,5)
y=(1,2,3,4,5,6,7,8,9,10)
status=[0,0,0,0,0,0,0,0,0,0]
new_status=[3,0,0,2,0,0,0,0,4,1]
ax = plt.subplot()
y16 = (y[0], y[5])
x16 = (x[0], x[5])
ax.plot(y16 ,x16, marker='o', color='blue')
y28 = (y[7], y[1])
x28 = (x[7], x[1])
ax.plot(y28, x28, marker='o', color='blue')
y37 = (y[2], y[6])
x37 = (x[2], x[6])
ax.plot(y37, x37, marker='o', color='blue')
y410 = (y[3], y[9])
x410 = (x[3], x[9])
ax.plot(y410, x410, marker='o', color='blue', )
y59 = (y[4], y[8])
x59 = (x[4], x[8])
ax.plot(y59, x59, marker='o', color='blue')
plt.show()

我想你可以画多次,一次是两个头和一条线,然后是一个头。我已经为每个新状态分配了不同的颜色,然后再次以不同的颜色绘制单个点

x=(1,2,3,4,5,1,2,3,4,5)
y=(1,2,3,4,5,6,7,8,9,10)
status=[0,0,0,0,0,0,0,0,0,0]
new_status=[3,0,0,2,0,0,0,0,4,1]
colours =['r', 'b', 'g', 'y', 'c'] 
y16 = (y[0], y[5])
x16 = (x[0], x[5])
plt.plot(y16 ,x16, marker='o', color=colours[new_status[0]])
plt.plot(y[5], x[5], marker='o', color=colours[new_status[5]])
y28 = (y[7], y[1])
x28 = (x[7], x[1])
plt.plot(y28, x28, marker='o', color=colours[new_status[7]])
plt.plot(y[1], x[1], marker='o', color=colours[new_status[1]])
y37 = (y[2], y[6])
x37 = (x[2], x[6])
plt.plot(y37, x37, marker='o', color=colours[new_status[2]])
plt.plot(y[6], x[6], marker='o', color=colours[new_status[6]])
y410 = (y[3], y[9])
x410 = (x[3], x[9])
plt.plot(y410, x410, marker='o', color=colours[new_status[3]])
plt.plot(y[9], x[9], marker='o', color=colours[new_status[9]])
y59 = (y[4], y[8])
x59 = (x[4], x[8])
plt.plot(y59, x59, marker='o', color=colours[new_status[4]])
plt.plot(y[8], x[8], marker='o', color=colours[new_status[8]])
plt.show()