使用线图 plt.plot()
plt.plot()只允许一种颜色。因此，您可以简单地遍历数据和颜色并分别绘制每个点。
import numpy as npimport matplotlib.pyplot as pltfig, ax = plt.subplots()data = np.array([[4.29488806,-5.34487081], [3.63116248,-2.48616998], [-0.56023222,-5.89586997], [-0.51538502,-2.62569576], [-4.08561754,-4.2870525 ], [-0.80869722,10.12529582]])colors = ['red','red','red','blue','red','blue']for xy, color in zip(data, colors): ax.plot(xy[0],xy[1],'o',color=color, picker=True)plt.show()
使用散点图 plt.scatter()
为了生成散点图，请使用
scatter。它有一个论点
c，它允许使用多种方式设置散点的颜色。
（a） 一种简单的方法是提供颜色列表。
colors = ['red','red','red','blue','red','blue']ax.scatter(data[:,0],data[:,1],c=colors,marker="o", picker=True)
（b） 另一个选择是提供数据列表，并使用颜色图将数据映射为颜色
colors = [0,0,0,1,0,1] #red is 0, blue is 1ax.scatter(data[:,0],data[:,1],c=colors,marker="o", cmap="bwr_r")
