覆盖tricontour图时的颜色栏/图例选项

2024-05-19 23:25:56 发布

您现在位置:Python中文网/ 问答频道 /正文

我已经创建了三个tricontour图,并覆盖了这些图,如下所示:

finalDf = pca_cont('%_helix')
x1 = list(finalDf['pc1'])
y1 = list(finalDf['pc2'])
z1 = list(finalDf['%_helix'])
zf1 = threshold (z1, 0.60)

finalDf2 = pca_cont('%_sheet')
x2 = list(finalDf2['pc1'])
y2 = list(finalDf2['pc2'])
z2 = list(finalDf2['%_sheet'])
zf2 = threshold (z2, 0.60)

finalDf3 = pca_cont('%_coil')
x3 = list(finalDf3['pc1'])
y3 = list(finalDf3['pc2'])
z3 = list(finalDf3['%_coil'])
zf3 = threshold (z3, 0.90)

plt.figure(figsize = (16,16))
plt.tricontour(x1,y1,zf1,500,cmap="Reds", alpha=.5)
plt.tricontour(x2,y2,zf2,500,cmap="Greens", alpha=.5)
plt.tricontour(x3,y3,zf3,500,cmap="Blues", alpha=.5)
plt.xlabel("Principal Component 1")
plt.ylabel("Principal Component 2")
plt.title("100 Dimension Embedding")
plt.colorbar()


plt.show()

plot

但我的问题是,我有三组颜色和不同的深度,所以当我使用plt.colorbar()添加颜色条时,它不会显示所有三种颜色。表示这三组数据的最佳方式是什么?我应该只画一个简单的图例,还是有办法用三个色条来表示三组数据

多谢各位


Tags: alphathreshold颜色pltlistcmappcahelix