python散点图自定义_Python Matplotlib散点图：根据条件指定颜色点

我有两个numpy数组，x和y，每个数组有7000个元素。我想绘制一个散点图，根据这些条件给每个点不同的颜色：-BLACK if x[i]<10.

-RED if x[i]>=10 and y[i]<=-0.5

-BLUE if x[i]>=10 and y[i]>-0.5

我试着创建一个与数据长度相同的列表，列表的颜色是我想分配给每个点的颜色，然后用循环绘制数据，但是运行它需要很长时间。这是我的代码：import numpy as np

import matplotlib.pyplot as plt

#color list with same length as the data

col=[]

for i in range(0,len(x)):

if x[i]<10:

col.append('k')

elif x[i]>=10 and y[i]<=-0.5:

col.append('r')

else:

col.append('b')

#scatter plot

for i in range(len(x)):

plt.scatter(x[i],y[i],c=col[i],s=5, linewidth=0)

#add horizontal line and invert y-axis

plt.gca().invert_yaxis()

plt.axhline(y=-0.5,linewidth=2,c='k')

在此之前，我尝试以相同的方式创建相同的颜色列表，但打印的数据没有循环：#scatter plot

plt.scatter(x,y,c=col,s=5, linewidth=0)

尽管这比for循环绘制数据要快得多，但一些分散的点显示错误的颜色。为什么不使用循环来绘制数据会导致某些点的颜色不正确？

我还尝试定义了三组数据，每种颜色一组，并分别将它们添加到绘图中。但这不是我要找的。

是否有方法在散点图参数中指定要用于每个点的颜色列表，以便不使用for循环？

注：这是我不使用for循环时得到的图(错误的一个)：

这个当我使用for循环时(正确)：