pcolormesh (または pcolor) を使用して、ポリゴンで構成される配列を作成できます。これらは任意の形状にすることができます。matshow や imshow のような通常の配列プロットは、軸に沿って常に一定のサイズになると思います。
n = 6
# generate some data
gdp = np.array(np.random.randint(10,500,n))
countries = np.array(['Country%i' % (i+1) for i in range(n)])
matr = np.random.randint(0,10,(n,n))
# get the x and y arrays
y = np.insert(gdp.cumsum(),0,0)
xx,yy = np.meshgrid(np.arange(n+1),y)
# plot the matrix
fig, axs = plt.subplots(figsize=(6,6))
axs.pcolormesh(xx,yy,matr.T, cmap=plt.cm.Reds, edgecolors='k')
axs.set_ylim(y.min(),y.max())
# set the yticks + labels
axs.set_yticks(y[1:] - np.diff(y) / 2)
axs.set_yticklabels(countries)
#set xticks + labels
axs.xaxis.set_ticks_position('top')
axs.set_xticks(np.arange(n)+0.5)
axs.set_xticklabels(np.arange(n))
高さは次のようにスケーリングされます。
print countries
['Country1' 'Country2' 'Country3' 'Country4' 'Country5' 'Country6']
print gdp
[421 143 134 388 164 420]