使用线图plt.plot()

plt.plot()只允许一种颜色。因此，您可以简单地循环数据和颜色，并单独绘制每个点。

import numpy as np
import matplotlib.pyplot as plt

fig, 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）**另一种选择是提供数据列表，并使用色彩Map表将数据Map到颜色

colors = [0,0,0,1,0,1] #red is 0, blue is 1
ax.scatter(data[:,0],data[:,1],c=colors,marker="o", cmap="bwr_r")

您必须设置plt.scatter的参数c，并列出所需的颜色：

import matplotlib.pylab as plt
import numpy as np

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']
plt.scatter(data[:,0],data[:,1],marker='o',c = colors)
plt.show()