c:调节不同类别散点颜色;
marker:调节标记样式;
zorder:调节函数画点所在层数,zorder整数越大,显示时越靠上。
参考:
plt.scatter 分类数据 legend 添加图例
Python使scatter函数画点在plot函数上层
plt.figure(figsize=(10, 10))
colors = ['b','r']
s = [0,1]
marker1 = ["^", "o"]
for index in range(2):
XOffset = df_SPI2.loc[df_SPI2['Result'] == index]['XOffset']
YOffset = df_SPI2.loc[df_SPI2['Result'] == index]['YOffset']
s[index] = plt.scatter(XOffset, YOffset, c=colors[index], cmap='brg', s=50, alpha=1, marker=marker1[index], linewidth=0,zorder=-index)
# plt.legend(result[index])
plt.legend((s[0],s[1]),('Good','Fail') ,loc = 'best')
plt.xlabel("XOffset")
plt.ylabel("YOffset")
plt.title('Scatter Plot')
直接用pandas绘制
参考:
pandas绘图官方文档
pandas.DataFrame.plot绘制散点图
用颜色代表价格两种标签(“Good”,“Fail”),jet预定义颜色表(参数cmap)
df = df_SPI2[["XOffset","YOffset","Result"]]
df.plot.scatter(x = "XOffset", y = "YOffset",c = "Result",cmap=plt.get_cmap("jet"),colorbar=True)