plt.scatter 中cmap参数详解

1.首先,cmap参数和c参数配合使用的。参数c可以是一个序列,如:plt.scatter(a,b,c=[b,r,b,r,b],s=80)

此时c的序列是一个颜色序列,除了上述的简洁写法,还可以使用RGB或RBGA:plt.scatter(a,b,c=[#f00,#0f0,#f00,#0f0,#f00],s=80)

上面的例子是将参数c设置为颜色序列,但是一般我们用来做分类的数据Y是整数0,1,2,.......,等等。此时就需要配合cmap参数来使用了。如:

a = np.array([1,2,3,4,5])
b= np.array([6,7,8,9,10])
c = np.array([0,1,2,3,4])
plt.scatter(a,b,c=c,s=80,cmap=plt.cm.Spectral)

其结果:

在这里需要注意的是,plt.cm.Spectral是一个颜色映射集,但是,并不是说0~5各代表了某个颜色。而是说,参数c的序列中出现了5个不同的值,然后为每一个值分配一个颜色。

如下例:

a = np.array([1,2,3,4,5])
b= np.array([6,7,8,9,10])
c = np.array([0,1,2,3,4])
plt.scatter(a,b,c=[0,1,0,1,0],s=80,cmap=plt.cm.Spectral)
经验分享 程序员 微信小程序 职场和发展