python - 如何更改循环/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]

然后当你制作'Events'时，设置color=y0colors，像这样:

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

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