Matplotlib散点图根据列表上的值更改颜色

Matplotlib散点图根据列表上的值更改颜色 使用线图

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")