如何更改循环/if 语句中散点图的颜色？ （Python）

Question

我试图做到这一点，如果事件的散点图位于硅藻土散点图之上，则图的颜色从蓝色变为红色。 （如果不是的话，同样从蓝色到黄色）。

我已经环顾四周很长一段时间了，但似乎不起作用。

代码如下：

for line in open('C:\...'):
    line = line.split()
    y = line[9]
    if y == "END":
        break
    x = line[10]
    if x == 'END':
        break
    z = line[11]
    if z == 'END':
    break
    x0.append(float(x))
    y0.append(float(y))
    z0.append(float(z))

for line in open('C:\...'):
    line=line.split()
    c = line[0]
    a = line[1]
    b = line[2]
    a0.append(float(a))
    b0.append(float(b))
    c0.append(float(c))
##    print c0
    Tave = average(c0)
    print Tave


fig = pylab.figure() 
ax = p3.Axes3D(fig) 
Diatomite = ax.scatter(a0,b0,c0, color='green')
Events = ax.scatter(x0, z0, y0, color='b')
pyplot.show() 

for value in y0[:]:
    if value > Tave:
        ax.scatter(color = 'red')
    else:
        ax.scatter(color = 'yellow')

scatter(color=colors)

---

任何帮助将不胜感激！ （请注意，我取出了文件的路径。）

Answer 1

看起来你几乎已经拥有它了。首先定义颜色，然后绘制散点图。因此，在 print Tave 之后，构建一个包含颜色的列表；我使用列表理解来执行此操作：

y0colors = ['red' if value > Tave else 'yellow' for value in y0]

然后，当您创建“事件”时，设置color=y0colors，如下所示：

Events = ax.scatter(x0, z0, y0, color=y0colors)

当然，根据您使用的要求，所有点都不会是蓝色的。 y0 值要么大于 Tave（“红色”），要么大于 Tave（“黄色”）。在定义散点图“事件”之后可能有一种方法可以做到这一点，但这种方法似乎更直接。