Colorbar внутри фигуры, попытался создать новые оси

Question

Я использую следующий код для создания контуров, похожих на это 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 также были помещены на эту ось. Я посмотрел на сайте, пытаясь понять это без удачи, поэтому любая помощь приветствуется.