Выберите размер фигур на цветовой шкале карты цветов

У меня есть этот код, который работает (я взял его с этого форума). Я хотел бы добавить возможность изменять размер фигур справа от цветовой шкалы. Кто-нибудь может мне помочь ?

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.ticker as ticker

X = np.random.uniform(-2, 2, 200)
Y = np.random.uniform(-2, 2, 200)
Z = X*np.exp(-X**2 - Y**2)

contour = plt.tricontourf(X, Y, Z, 100, cmap="bwr")

def fmt(x, pos):
    a, b = '{:.2e}'.format(x).split('e')
    b = int(b)
    return r'${} \times 10^{{{}}}$'.format(a, b)

cbar = plt.colorbar(contour, format=ticker.FuncFormatter(fmt))
cbar.set_label('Colorbar label', size = 15)

plt.show()

У меня было это изображение, чтобы помочь:

Большое спасибо !