图形内的颜色栏,尝试创建新轴

时间:2018-11-06 01:59:25

标签: python matplotlib colorbar

我正在使用下面的代码生成看起来像picture的轮廓。我遇到的问题是,我试图将颜色条放在图形或轮廓内(如果可以的话),但我没有设法正确处理它。

r = np.linspace(0.1, 50, 100)

t = np.linspace(0, 3, 100)

X, Y = np.meshgrid(r, t)

Z1 = np.zeros((100,100))
Z2 = np.zeros((100,100))

for i in range(100):
    for j in range(100):
        cacheTuple = growthMap(Y[i,j],0.01, 0.01, 0.01, 0.01, X[i,j])
        Z1[i,j] = cacheTuple[0]
        Z2[i,j] = cacheTuple[1]

cs = plt.contourf(X, Y, Z1, 100, cmap=plt.get_cmap('gist_rainbow_r'))
plt.contour(X, Y, Z2, [1, 5, 10, 15, 20])
plt.annotate(r'$\xi = 0.01$', (0.04,0.08), xycoords = 'axes fraction')
plt.annotate(r'$\alpha$ = St = 0.001', (0.04,0.03), xycoords = 'axes fraction')


csb = plt.colorbar(cs, ticks=[-2, -1, 0, 1, 2, 3, 4])
csb.set_label(r'log($M/M_E$)')
plt.ylabel(r'$t_0$[Myr]')
plt.xlabel(r'$r_0$[AU]')
plt.savefig('Growth_map1.png', dpi=500)

我尝试创建新轴并在其上放置颜色条,但是当我这样做时,x和y标签也被放置在该轴上。我在网站上四处张望,试图解决问题,所以没有任何帮助,谢谢。

0 个答案:

没有答案