使用matplotlib更改colorbar图中框的宽度

Question

我有一个非常简单的颜色条（为此我删除了图例）。我的数据是结构化的，因此行数多于列数。我想让每个方框都更宽，以使宽高比的宽高比接近1：1，但在属性文档中没有找到任何可以做到这一点。

import matplotlib.pyplot as plt
import numpy as np
from matplotlib import cm

fig, ax = plt.subplots()

data = np.random.choice([0, 1], size=[2000,20])


cax = ax.imshow(data, interpolation='nearest', cmap=cm.coolwarm)
ax.axes.get_xaxis().set_visible(False)
ax.axes.get_yaxis().set_visible(False)

plt.show()