如何调整热图单元的高度/宽度？

Question

我用以下代码在python中绘制了一个简单的热图。我将如何使用matplotlib调整单个热图单元的高度/宽度？​​

def HeatMap(data):

    # plot the figure

    figure = plt.figure()
    sub_figure = figure.add_subplot(111)
    heatmap = sub_figure.imshow(data, interpolation='nearest',cmap='jet')

    # add a color bar

    cbar = figure.colorbar(ax=sub_figure, mappable=one_box, orientation='horizontal')
    cbar.set_label('Scores')

    plt.show()

例如，我尝试过figure = plt.figure(figsize=(10,10))，但是它会增加/减少整个图形而不是单个单元格。到目前为止，我有这个：

我希望这些细胞的高度较小，以便压扁。