mpl.colors.listedcolormap
时间: 2023-04-23 10:04:29 浏览: 250
mpl.colors.listedcolormap是Matplotlib中的一个函数,用于创建一个基于离散值的颜色映射。它可以将一组离散的数据值映射到一组预定义的颜色中,每个颜色对应一个数据值。这个函数可以用于可视化分类数据或者离散数据的分布情况。
相关问题
cm_light = mpl.colors.ListedColormap(['#A0FFA0', '#FFA0A0', '#A0A0FF']) cm_dark = mpl.colors.ListedColormap(['g', 'r', 'b'])
这段代码定义了两个颜色映射(colormap):`cm_light` 和 `cm_dark`。`cm_light` 是一个由三种颜色组成的映射,分别是浅绿色(#A0FFA0)、浅红色(#FFA0A0)和浅蓝色(#A0A0FF)。`cm_dark` 是一个由三种颜色组成的映射,分别是绿色(g)、红色(r)和蓝色(b)。这些颜色映射可以用于在 matplotlib 中设置图表中的颜色。
def drawPlot(title,x_train,x_test,y_train,y_test): N,M=500,500 x1_min,x2_min=x_train.min() x1_max,x2_max=x_train.max() t1=np.linspace(x1_min,x1_max,N) t2=np.linspace(x2_min,x2_max,M) x1,x2=np.meshgrid(t1,t2) x_show=np.stack((x1.flat,x2.flat),axis=1) y_hat=model.predict(x_show) y_hat=y_hat.reshape(x1.shape) cm_light=mpl.colors.ListedColormap(['#F2F2C2','#E00E00','#FFFFFF']) cm_dark=mpl.colors.ListedColormap(['#FFFFFF','#000000','#CCCCCC']) plt.figure(figsize=(3,2.5),facecolor='w') plt.contour(x1,x2,y_hat,colors='k',levels=[0,1],antialiased=True,linewidths=1) plt.pcolormesh(x1,x2,y_hat,cmap=cm_light) plt.scatter(x_train[0],x_train[1],c=y_train,s=20,edgecolors='k',cmap=cm_dark,label='训练集') plt.scatter(x_test[0],x_test[1],c=y_test,s=80,marker='*',edgecolors='k',cmap=cm_dark,label='测试集') plt.xlabel('花萼长度',fontsize=9) plt.ylabel('花萼宽度',fontsize=9) plt.xlim(x1_min,x1_max) plt.ylim(x2_min,x2_max) plt.grid(b=True,ls=':',color='#606060') plt.suptitle(title,fontsize=8) plt.tight_layout(1,rect=(0,0,1,0.94)) plt.show()报错TypeError: tight_layout() takes 0 positional arguments but 1 positional argument (and 1 keyword-only argument) were given
这个错误是因为 tight_layout() 这个函数需要传入一个子图布局的矩形参数,但是在这里没有给定。你可以将代码中的 plt.tight_layout(1,rect=(0,0,1,0.94)) 修改为 plt.tight_layout(rect=(0,0,1,0.94)),这样就可以解决这个问题了。
阅读全文