如何在loop / if语句中更改散点图的颜色？ （蟒蛇）

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（'red'）或不大（'yellow'）。在定义散点图“事件”之后，可能有一种方法可以做到这一点，但这种方式似乎更直接。