python散点图属性&用不同颜色展示不同的分类结果(分类着色)

对于画多重颜色的散点图,将不同颜色的数据用labels分开就好。在这里举一个例子:
假如需要四种颜色:
1. 用label_com存储四个标签的名字,
2. 用colors存储四种颜色
3. 在data中找到标签为对应label_com的labeldata.loc[data['Label'] == index],后面的price和number是我的数据 data其中的两个列。
4. 绘制散点图,用for index in range(4)遍历所有颜色,c=color[index]取到每种颜色。后面的部分是绘制legend所用。

fig1 = plt.figure(1,figsize=(6,4))

colors = ['b','g','r','orange']
Label_Com = ['Component 1','Component 2','Component 3','Component 4']
for index in range(4):
    Price = data.loc[data['Label'] == index]['Price']
    Index = data.loc[data['Label'] == index]['Number']
    plt.scatter(Index, Price, c=colors[index], cmap='brg', s=40, alpha=0.2, marker='8', linewidth=0)  

plt.ylim(0.01,0.09)

ax = fig1.gca()
for label in ax.xaxis.get_ticklabels():
    label.set_rotation(30)
plt.xlabel('Time')
plt.ylabel('Price')
#added this to get the legend to work
handles,labels = ax.get_legend_handles_labels()
ax.legend(handles, labels = Label_Com, loc='upper right')

plt.show()

cmap:可以用来调整颜色风格 详见https://matplotlib.org/examples/color/colormaps_reference.html
散点图plt.scatter的一些属性:
x:指定散点图的x轴数据;
y:指定散点图的y轴数据;
s:指定散点图点的大小,默认为20,通过传入新的变量,实现气泡图的绘制;
c:可用于不同类别的颜色,指定散点图点的颜色,默认为蓝色;
marker:指定散点图点的形状,默认为圆形;
cmap:指定色图,只有当c参数是一个浮点型的数组的时候才起作用;
alpha:设置散点的透明度;
linewidths:设置散点边界线的宽度;
edgecolors:设置散点边界线的颜色;

展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 编程工作室 设计师: CSDN官方博客
应支付0元
点击重新获取
扫码支付

支付成功即可阅读